yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-15 04:26:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix #11882 upgrade MLflow to 2.x (#13299)

Browse Source
This commit is contained in:
vanshika18 2023-09-26 11:04:15 +05:30 committed by GitHub
parent 3769a9d105
commit 6f92c3be1f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ingestion/setup.py
View File

@ -197,7 +197,7 @@ plugins: Dict[str, Set[str]] = {
"kinesis": {VERSIONS["boto3"]},
"ldap-users": {"ldap3==2.9.1"},
"looker": {"looker-sdk>=22.20.0", "lkml~=1.3"},
"mlflow": {"mlflow-skinny~=1.30", "alembic~=1.10.2"},
"mlflow": {"mlflow-skinny>=2.3.0", "alembic~=1.10.2"},
"mongo": {"pymongo~=4.3", VERSIONS["pandas"]},
"couchbase": {"couchbase~=4.1"},
"mssql": {"sqlalchemy-pytds~=0.3"},

2
ingestion/src/metadata/ingestion/source/mlmodel/mlflow/connection.py
View File

@ -47,7 +47,7 @@ def test_connection(
of a metadata workflow or during an Automation Workflow
"""
test_fn = {"GetModels": client.list_registered_models}
test_fn = {"GetModels": client.search_registered_models}
test_connection_steps(
metadata=metadata,

6
ingestion/src/metadata/ingestion/source/mlmodel/mlflow/metadata.py
View File

@ -69,7 +69,7 @@ class MlflowSource(MlModelServiceSource):
"""
List and filters models from the registry
"""
for model in cast(RegisteredModel, self.client.list_registered_models()):
for model in cast(RegisteredModel, self.client.search_registered_models()):
if filter_by_mlmodel(
self.source_config.mlModelFilterPattern, mlmodel_name=model.name
):
@ -92,8 +92,8 @@ class MlflowSource(MlModelServiceSource):
self.status.failed(
StackTraceError(
name=model.name,
error="Invalid Version",
stack_trace=f"Cannot find latest version from version list {model.latest_versions}",
error="Version not found",
stack_trace=f"Unable to ingest model {model.name} due to missing version from version list {model.latest_versions}", # pylint: disable=line-too-long
)
)
continue