wehub-resource-sync
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from functools import wraps
|
|
from typing import Any, Callable, TypeVar, cast
|
|
|
|
from mlflow.exceptions import MlflowException
|
|
from mlflow.protos.databricks_pb2 import FEATURE_DISABLED
|
|
|
|
F = TypeVar("F", bound=Callable[..., Any])
|
|
|
|
|
|
def filestore_not_supported(func: F) -> F:
|
|
"""
|
|
Decorator for FileStore methods that are not supported.
|
|
|
|
This decorator wraps methods to raise a helpful error message when
|
|
SQL-backend-only features are called on a FileStore instance.
|
|
|
|
Returns:
|
|
A wrapped function that raises MlflowException when called.
|
|
"""
|
|
|
|
@wraps(func)
|
|
def wrapper(self, *args, **kwargs):
|
|
raise MlflowException(
|
|
f"{func.__name__} is not supported with FileStore. "
|
|
f"This feature requires a SQL-based tracking backend "
|
|
f"(e.g., SQLite, PostgreSQL, MySQL). Please configure MLflow "
|
|
f"with a SQL backend using --backend-store-uri. "
|
|
f"For SQLite setup instructions, see: "
|
|
f"https://mlflow.org/docs/latest/self-hosting/architecture/tracking-server/#configure-server",
|
|
error_code=FEATURE_DISABLED,
|
|
)
|
|
|
|
return cast(F, wrapper)
|