# __init__.py
from mlflow.store.artifact.artifact_repository_registry import register_artifact_repository
from .hf_artifact_repo import HuggingFaceArtifactRepository

def register_hf_artifact_repository():
    """Register HuggingFace Hub as an artifact repository"""
    register_artifact_repository("hf", HuggingFaceArtifactRepository)
    print("✅ HuggingFace Artifact Repository registered!")

# Auto-register when imported
register_hf_artifact_repository()