Merge pull request #150 from splicemachine/DBAAS-5518

DBAAS-5518: live model status route
splicemachine · May 19, 2021 · c9fc393 · c9fc393
2 parents 7253c45 + 7214289
commit c9fc393
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/splicemachine/mlflow_support/mlflow_support.py b/splicemachine/mlflow_support/mlflow_support.py
@@ -1055,11 +1055,13 @@ def _get_deployed_models() -> PandasDF:
     :return: Pandas df
     """
 
-    return mlflow._splice_context.df(
-        """
-        SELECT * FROM MLMANAGER.LIVE_MODEL_STATUS
-        """
-    ).toPandas()
+    request = requests.get(
+        get_jobs_uri(mlflow.get_tracking_uri() or get_pod_uri('mlflow', 5003, _testing=_TESTING)) + "/api/rest/deployments",
+        auth=mlflow._basic_auth
+    )
+    if not request.ok:
+        raise SpliceMachineException(request.text)
+    return PandasDF(dict(request.json()))
 
 
 def apply_patches():