wehub-resource-sync
26 lines
668 B
Python
26 lines
668 B
Python
from functools import lru_cache
|
|
|
|
from mlflow.assistant.config import AssistantConfig
|
|
|
|
|
|
@lru_cache(maxsize=100)
|
|
def get_project_path(experiment_id: str) -> str | None:
|
|
"""Get the project path for a given experiment ID.
|
|
|
|
Args:
|
|
experiment_id: The experiment ID to look up.
|
|
|
|
Returns:
|
|
The project path if found, None otherwise.
|
|
"""
|
|
config = AssistantConfig.load()
|
|
return config.get_project_path(experiment_id)
|
|
|
|
|
|
def clear_project_path_cache() -> None:
|
|
"""Clear the project path cache to pick up config changes."""
|
|
get_project_path.cache_clear()
|
|
|
|
|
|
__all__ = ["get_project_path", "clear_project_path_cache", "AssistantConfig"]
|