gitea_admin/sientia-dataops-laborious-legacy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

QTZPOC-20

Browse Source 
QTZPOC-20: Make metadata parameter optional in check_artifact_exists method.
This commit is contained in:
Vitor Pimentel dos Santos
2026-09-01 13:54:51 -03:00
parent 30ff378cb7
commit 086b65c45f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
laborious/utils/repository/model_repository.py
View File

@@ -256,7 +256,7 @@ class MLFlowRepository(SientiaMonitoring):
return run_info.data.params return run_info.data.params
def check_artifact_exists( def check_artifact_exists(
self, run_id: str, artifact_path: str, metadata: dict[str, Any] self, run_id: str, artifact_path: str, metadata: dict[str, Any] | None = None
) -> bool: ) -> bool:
""" """
Check if an artifact exists in the MLflow Model Registry. Check if an artifact exists in the MLflow Model Registry.
@@ -271,6 +271,9 @@ class MLFlowRepository(SientiaMonitoring):
run_id (str): Run identifier to inspect. run_id (str): Run identifier to inspect.
artifact_path (str): Path to the artifact to check, relative to the artifact_path (str): Path to the artifact to check, relative to the
run's artifact root, e.g. 'data_model' or 'data_model/code'. run's artifact root, e.g. 'data_model' or 'data_model/code'.
metadata (dict[str, Any] | None): Metadata used for structured
logging. Optional, so callers whose own metadata is optional —
`download_artifact_code_dir` — can forward it as is.
Returns: Returns:
bool: True if the artifact exists, False otherwise. bool: True if the artifact exists, False otherwise.