from absolute to relative import for build pytest

projectmesa · Aug 2, 2024 · 3cf1357 · 3cf1357
1 parent 5a33cba
commit 3cf1357
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/tests/pandas/test_grid_pandas.py b/tests/pandas/test_grid_pandas.py
@@ -4,11 +4,12 @@
 import typeguard as tg
 
 from mesa_frames import GridPandas, ModelDF
-from tests.pandas.test_agentset_pandas import (
+
+from ..pandas.test_agentset_pandas import (
     ExampleAgentSetPandas,
     fix1_AgentSetPandas,
 )
-from tests.polars.test_agentset_polars import (
+from ..polars.test_agentset_polars import (
     ExampleAgentSetPolars,
     fix2_AgentSetPolars,
 )

diff --git a/tests/test_agents.py b/tests/test_agents.py
@@ -7,12 +7,13 @@
 from mesa_frames import AgentsDF, ModelDF
 from mesa_frames.abstract.agents import AgentSetDF
 from mesa_frames.types_ import AgentMask
-from tests.pandas.test_agentset_pandas import (
+
+from .pandas.test_agentset_pandas import (
     ExampleAgentSetPandas,
     fix1_AgentSetPandas,
     fix2_AgentSetPandas,
 )
-from tests.polars.test_agentset_polars import (
+from .polars.test_agentset_polars import (
     ExampleAgentSetPolars,
     fix2_AgentSetPolars,
 )