diff --git a/.github/workflows/test-python.yaml b/.github/workflows/test-python.yaml
index 61f6c82f45..2515f3dece 100644
--- a/.github/workflows/test-python.yaml
+++ b/.github/workflows/test-python.yaml
@@ -23,7 +23,7 @@ jobs:
           src: sdk/
 
       - name: Install dependencies
-        run: pip install pytest python-dateutil urllib3 kubernetes
+        run: pip install pytest python-dateutil urllib3 kubernetes ./sdk/python
 
       - name: Run unit test for training sdk
         run: pytest ./sdk/python/kubeflow/training/api/training_client_test.py
\ No newline at end of file
diff --git a/sdk/python/kubeflow/storage_init_container/hugging_face.py b/sdk/python/kubeflow/storage_init_container/hugging_face.py
index 68c39dbaa1..c136b265d2 100644
--- a/sdk/python/kubeflow/storage_init_container/hugging_face.py
+++ b/sdk/python/kubeflow/storage_init_container/hugging_face.py
@@ -1,5 +1,5 @@
-from abstract_model_provider import modelProvider
-from abstract_dataset_provider import datasetProvider
+from kubeflow.storage_init_container.abstract_model_provider import modelProvider
+from kubeflow.storage_init_container.abstract_dataset_provider import datasetProvider
 from dataclasses import dataclass, field
 from typing import Literal
 from urllib.parse import urlparse
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
index 6654900a98..093fbeea27 100644
--- a/sdk/python/setup.py
+++ b/sdk/python/setup.py
@@ -37,6 +37,8 @@
     "einops>=0.7.0",
     "transformers_stream_generator>=0.0.4",
     "peft>=0.7.0",
+    "datasets>=2.15.0"
+    "huggingface-hub>=0.19.4"
 ]
 
 setuptools.setup(