adding validation for model uri

deepanker13 · Dec 12, 2023 · a017b21 · a017b21
1 parent eb7c9d0
commit a017b21
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sdk/python/kubeflow/storage_init_container/hugging_face.py b/sdk/python/kubeflow/storage_init_container/hugging_face.py
@@ -27,6 +27,8 @@ def __post_init__(self):
         # Custom checks or validations can be added here
         if self.transformer_type not in TRANSFORMER_TYPES:
             raise ValueError("transformer_type must be one of %s", TRANSFORMER_TYPES)
+        if self.model_uri is None:
+            raise ValueError("model_uri cannot be none.")
 
 
 @dataclass