fix: comparing DFs

projectmesa · Oct 1, 2024 · 2806466 · 2806466
1 parent 284163b
commit 2806466
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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