triggerflow 0.1.7__tar.gz → 0.1.8__tar.gz

{triggerflow-0.1.7 → triggerflow-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triggerflow
-Version: 0.1.7
+Version: 0.1.8
 Summary: Utilities for ML models targeting hardware triggers
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License

{triggerflow-0.1.7 → triggerflow-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "triggerflow"
-version = "0.1.7"
+version = "0.1.8"
 description = "Utilities for ML models targeting hardware triggers"
 readme = "README.md"
 requires-python = ">=3.10"

{triggerflow-0.1.7 → triggerflow-0.1.8}/src/triggerflow/mlflow_wrapper.py

@@ -15,7 +15,8 @@ def setup_mlflow(mlflow_uri: str = None,
                  run_name: str = None, 
                  experiment_id: str = None,
                  run_id: str = None,
-                 save_env_file=True
+                 save_env_file: bool = True,
+                 auto_configure: bool = False
                  ):
 
     # Set the MLflow tracking URI
@@ -110,13 +111,17 @@ def setup_mlflow(mlflow_uri: str = None,
             f.write(f"MLFLOW_EXPERIMENT_ID={experiment_id}\n")
             f.write(f"MLFLOW_RUN_ID={run_id}\n")
 
+            if auto_configure:
+                print("Auto_configure is set to true. Exporting AUTO_CONFIGURE=true")
+                f.write(f"AUTO_CONFIGURE=true\n")
+
     return experiment_name, run_name, experiment_id, run_id
 
-if os.getenv("MLFLOW_TRACKING_PASSWORD") is not None and os.getenv("MLFLOW_TRACKING_USERNAME") is not None and os.getenv('CI') == 'true':
-    print("Setup mlflow run")
+if os.getenv("AUTO_CONFIGURE") == "true":
+    print("AUTO_CONFIGURE is true and running in CI environment. Setting up mlflow...")
     setup_mlflow()
 else:
-    print("MLFLOW_TRACKING_PASSWORD and MLFLOW_TRACKING_USERNAME not set. Skipping mlflow run setup")
+    print("AUTO_CONFIGURE is not set. Skipping mlflow run setup")
 
 class MLflowWrapper(mlflow.pyfunc.PythonModel):
     """PyFunc wrapper for TriggerModel; backend can be set at runtime."""

{triggerflow-0.1.7 → triggerflow-0.1.8}/src/triggerflow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triggerflow
-Version: 0.1.7
+Version: 0.1.8
 Summary: Utilities for ML models targeting hardware triggers
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License