fail on unmet requirements.txt

DagsHub · Jul 24, 2024 · 064094b · 064094b
1 parent 6c23576
commit 064094b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/label_studio_ml/api.py b/label_studio_ml/api.py
@@ -42,7 +42,10 @@ def _configure():
     dagshub.init(*args['repo'].split('/')[::-1])  # user-level privileged auth token
 
     req_path = mlflow.pyfunc.get_model_dependencies(f'models:/{args["model"]}/{args["version"]}')
-    shell(f'yes | uv pip install -r {req_path}')
+    try:
+        shell(f'yes | uv pip install -r {req_path}')
+    except:
+        raise ValueError("Failed to install requirements.txt.")
 
     mlflow_model = mlflow.pyfunc.load_model(f'models:/{args["model"]}/{args["version"]}')