Fucius's picture
Upload 422 files
2eafbc4 verified
from typing import Tuple
from inference.core.entities.types import DatasetID, VersionID
from inference.core.exceptions import InvalidModelIDError
def get_model_id_chunks(model_id: str) -> Tuple[DatasetID, VersionID]:
model_id_chunks = model_id.split("/")
if len(model_id_chunks) != 2:
raise InvalidModelIDError(f"Model ID: `{model_id}` is invalid.")
return model_id_chunks[0], model_id_chunks[1]