don't fail on vispy import

DamCB · Feb 14, 2024 · 998eac3 · 998eac3
1 parent ffb1603
commit 998eac3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/tyssue/draw/__init__.py b/src/tyssue/draw/__init__.py
@@ -2,7 +2,12 @@
 from .ipv_draw import sheet_view as sheet_view_3d  # noqa
 from .plt_draw import create_gif, plot_forces, quick_edge_draw  # noqa
 from .plt_draw import sheet_view as sheet_view_2d  # noqa
-from .vispy_draw import sheet_view as sheet_view_vispy  # noqa
+
+try:
+    from .vispy_draw import sheet_view as sheet_view_vispy  # noqa
+except OSError:
+    print("vispy  won't work")
+    sheet_view_vispy = None
 
 
 def sheet_view(sheet, coords=["x", "y", "z"], ax=None, mode="2D", **draw_specs_kw):