Spaces:
Running
on
Zero
Running
on
Zero
File size: 419 Bytes
2eafbc4 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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]
|