make mesa runable without some dependencies

projectmesa · Jan 9, 2024 · 757993f · 757993f
1 parent ee060de
commit 757993f
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 6 deletions.
diff --git a/mesa/datacollection.py b/mesa/datacollection.py
@@ -38,7 +38,10 @@
 import types
 from functools import partial
 
-import pandas as pd
+try:
+    import pandas as pd
+except ImportError:
+    pass
 
 
 class DataCollector:

diff --git a/mesa/space.py b/mesa/space.py
@@ -27,7 +27,10 @@
 from typing import Any, Callable, TypeVar, Union, cast, overload
 from warnings import warn
 
-import networkx as nx
+try:
+    import networkx as nx
+except ImportError:
+    pass
 import numpy as np
 import numpy.typing as npt
 

diff --git a/mesa/visualization/__init__.py b/mesa/visualization/__init__.py
@@ -1,4 +1,7 @@
-from mesa_viz_tornado.ModularVisualization import *  # noqa
-from mesa_viz_tornado.modules import *  # noqa
-from mesa_viz_tornado.UserParam import *  # noqa
-from mesa_viz_tornado.TextVisualization import *  # noqa
+try:
+    from mesa_viz_tornado.ModularVisualization import *  # noqa
+    from mesa_viz_tornado.modules import *  # noqa
+    from mesa_viz_tornado.UserParam import *  # noqa
+    from mesa_viz_tornado.TextVisualization import *  # noqa
+except ImportError:
+    pass