guard viz import

capitalone · Jul 15, 2024 · c5aff00 · c5aff00
1 parent 0fa3d26
commit c5aff00
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rubicon_ml/intake_rubicon/viz.py b/rubicon_ml/intake_rubicon/viz.py
@@ -1,6 +1,5 @@
 from rubicon_ml import __version__
 from rubicon_ml.intake_rubicon.base import VizDataSourceMixin
-from rubicon_ml.viz import ExperimentsTable
 
 
 class ExperimentsTableDataSource(VizDataSourceMixin):
@@ -18,5 +17,8 @@ def __init__(self, metadata=None, **catalog_data):
 
     def _get_schema(self):
         """Creates an Experiments Table visualization and sets it as the visualization object attribute"""
+        from rubicon_ml.viz import ExperimentsTable
+
         self._visualization_object = ExperimentsTable(**self._catalog_data)
+
         return super()._get_schema()