From 086b65c45f25b0d5a1b2d0e7f5f93b724b8eb163 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Tue, 1 Sep 2026 13:54:51 -0300 Subject: [PATCH] QTZPOC-20 QTZPOC-20: Make metadata parameter optional in check_artifact_exists method. --- laborious/utils/repository/model_repository.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 76fbb4e..15b4a10 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -256,7 +256,7 @@ class MLFlowRepository(SientiaMonitoring): return run_info.data.params 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: """ Check if an artifact exists in the MLflow Model Registry. @@ -271,6 +271,9 @@ class MLFlowRepository(SientiaMonitoring): run_id (str): Run identifier to inspect. artifact_path (str): Path to the artifact to check, relative to the 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: bool: True if the artifact exists, False otherwise.