From 2806466b0bc545a790c89bc95b11f14ee5ef3af4 Mon Sep 17 00:00:00 2001
From: Adam Amer <136176500+adamamer20@users.noreply.github.com>
Date: Tue, 1 Oct 2024 12:32:10 +0200
Subject: [PATCH] fix: comparing DFs

---
 examples/sugarscape_ig/performance_comparison.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/sugarscape_ig/performance_comparison.py b/examples/sugarscape_ig/performance_comparison.py
index c593498e..1a27bcba 100644
--- a/examples/sugarscape_ig/performance_comparison.py
+++ b/examples/sugarscape_ig/performance_comparison.py
@@ -184,8 +184,8 @@ def plot_and_print_benchmark(
 
 
 def polars_equality_check(a: SugarscapePolars, b: SugarscapePolars):
-    assert_frame_equal(a.agents, b.agents, check_row_order=False)
-    assert_frame_equal(a.space, b.space, check_row_order=False)
+    assert_frame_equal(a.space.agents, b.space.agents, check_row_order=False)
+    assert_frame_equal(a.space.cells, b.space.cells, check_row_order=False)
     return True