Skip to content
This repository was archived by the owner on Apr 15, 2022. It is now read-only.

Commit c9fc393

Browse files
author
Ben Epstein
authored
Merge pull request #150 from splicemachine/DBAAS-5518
DBAAS-5518: live model status route
2 parents 7253c45 + 7214289 commit c9fc393

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

splicemachine/mlflow_support/mlflow_support.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1055,11 +1055,13 @@ def _get_deployed_models() -> PandasDF:
10551055
:return: Pandas df
10561056
"""
10571057

1058-
return mlflow._splice_context.df(
1059-
"""
1060-
SELECT * FROM MLMANAGER.LIVE_MODEL_STATUS
1061-
"""
1062-
).toPandas()
1058+
request = requests.get(
1059+
get_jobs_uri(mlflow.get_tracking_uri() or get_pod_uri('mlflow', 5003, _testing=_TESTING)) + "/api/rest/deployments",
1060+
auth=mlflow._basic_auth
1061+
)
1062+
if not request.ok:
1063+
raise SpliceMachineException(request.text)
1064+
return PandasDF(dict(request.json()))
10631065

10641066

10651067
def apply_patches():

0 commit comments

Comments
 (0)