yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-16 04:57:11 +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"]}, "kinesis": {VERSIONS["boto3"]},
"ldap-users": {"ldap3==2.9.1"}, "ldap-users": {"ldap3==2.9.1"},
"looker": {"looker-sdk>=22.20.0", "lkml~=1.3"}, "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"]}, "mongo": {"pymongo~=4.3", VERSIONS["pandas"]},
"couchbase": {"couchbase~=4.1"}, "couchbase": {"couchbase~=4.1"},
"mssql": {"sqlalchemy-pytds~=0.3"}, "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 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( test_connection_steps(
metadata=metadata, 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 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( if filter_by_mlmodel(
self.source_config.mlModelFilterPattern, mlmodel_name=model.name self.source_config.mlModelFilterPattern, mlmodel_name=model.name
): ):
@ -92,8 +92,8 @@ class MlflowSource(MlModelServiceSource):
self.status.failed( self.status.failed(
StackTraceError( StackTraceError(
name=model.name, name=model.name,
error="Invalid Version", error="Version not found",
stack_trace=f"Cannot find latest version from version list {model.latest_versions}", 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 continue