command to create presigned URL for MLFlow

This commit is contained in:
2026-05-27 10:52:08 -04:00
parent e1c8d6574f
commit 58681cef82
6 changed files with 64 additions and 6 deletions

View File

@@ -1,5 +1,6 @@
from __future__ import annotations
import os
from dataclasses import dataclass
from typing import Any, Protocol
@@ -40,6 +41,8 @@ class MlflowTracker:
if cfg.mlflow.mode is MlflowMode.disabled:
return NoopTracker()
os.environ.setdefault("MLFLOW_SUPPRESS_PRINTING_URL_TO_STDOUT", "true")
try:
import mlflow
except ImportError as e: