QTZPOC-20
QTZPOC-20: Make metadata parameter optional in check_artifact_exists method.
This commit is contained in:
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user