guard viz import and fix prefect dependency (#463)

* guard viz import
* update prefect
* add missing prefect import
* not on 3.11
* just use pip
* only in setup
* use proper syntax

docs/docs-environment.yml

@@ -24,7 +24,11 @@ dependencies:
   - jsonpath-ng<=1.6.1,>=1.5.3
   - numpy<=2.0.0,>=1.22.0
   - pandas<=2.2.2,>=1.0.0
-  - prefect<=2.19.7,>=2.16.5
+  - prefect<=2.19.8,>=2.16.5
   - pyarrow<=16.1.0,>=14.0.1
   - PyYAML<=6.0.1,>=5.4.0
   - scikit-learn<=1.5.0,>=0.22.0
+
+  - pip:
+    # additional prefect extras
+    - backports.strenum<=1.3.1,>=1.3.1;python_version<'3.11'

environment.yml

@@ -16,7 +16,7 @@ dependencies:
   - scikit-learn<=1.5.0,>=0.22.0
 
   # for prefect extras
-  - prefect<=2.19.7,>=2.16.5
+  - prefect<=2.19.8,>=2.16.5
 
   # for s3fs extras
   - s3fs<=2024.6.1,>=0.4
@@ -51,3 +51,7 @@ dependencies:
   # for edgetest
   - edgetest
   - edgetest-conda
+
+  - pip:
+    # additional prefect extras
+    - backports.strenum<=1.3.1,>=1.3.1;python_version<'3.11'

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()

setup.cfg

@@ -40,7 +40,8 @@ install_requires =
 
 [options.extras_require]
 prefect = 
-	prefect<=2.19.7,>=2.16.5
+	backports.strenum<=1.3.1,>=1.3.1;python_version<'3.11'
+	prefect<=2.19.8,>=2.16.5
 s3 = 
 	s3fs<=2024.6.1,>=0.4
 ui =