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

Commit 858a699

Browse files
author
Epstein
authored
DBAAS-3481: update because mlmanager schema merged (#36)
1 parent a4e57c5 commit 858a699

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

splicemachine/ml/management.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -186,9 +186,8 @@ def _get_model_type(pipeline_or_model) -> ModelType:
186186
class MLManager(MlflowClient):
187187
"""
188188
A class for managing your MLFlow Runs/Experiments
189-
"""
190-
#FIXME: THIS WILL NEED TO BE MLMANAGER AFTER DBAAS-3254
191-
MLMANAGER_SCHEMA = 'SPLICE'
189+
"""
190+
MLMANAGER_SCHEMA = 'MLMANAGER'
192191

193192
ARTIFACT_INSERT_SQL = f'INSERT INTO {MLMANAGER_SCHEMA}.ARTIFACTS (run_uuid, name, "size", "binary") VALUES (?, ?, ?, ?)'
194193
ARTIFACT_RETRIEVAL_SQL = 'SELECT "binary" FROM ' + f'{MLMANAGER_SCHEMA}.' + 'ARTIFACTS WHERE name=\'{name}\' ' \

0 commit comments

Comments
 (0)