import fixes

sdk/python/kubeflow/storage_init_container/hugging_face.py

@@ -36,7 +36,7 @@ class HuggingFaceModelParams:
 
     def __post_init__(self):
         # Custom checks or validations can be added here
-        if self.model_uri == "":
+        if self.model_uri == "" or self.model_uri is None:
             raise ValueError("model_uri cannot be empty.")
 
     @property
@@ -88,7 +88,7 @@ class HfDatasetParams:
 
     def __post_init__(self):
         # Custom checks or validations can be added here
-        if self.repo_id is None:
+        if self.repo_id == "" or self.repo_id is None:
             raise ValueError("repo_id is None")
 
     @property

sdk/python/kubeflow/storage_init_container/s3.py

@@ -1,4 +1,4 @@
-from .abstract_dataset_provider import datasetProvider
+from kubeflow.storage_init_container.abstract_dataset_provider import datasetProvider
 from dataclasses import dataclass, field
 import json, os
 import boto3

sdk/python/setup.py

@@ -37,8 +37,7 @@
     "einops>=0.7.0",
     "transformers_stream_generator>=0.0.4",
     "peft>=0.7.0",
-    "datasets>=2.15.0"
-    "huggingface-hub>=0.19.4"
+    "datasets>=2.15.0" "huggingface-hub>=0.19.4",
 ]
 
 setuptools.setup(