fixing github workflow

kubeflow · Jan 4, 2024 · cd811cd · cd811cd
1 parent 23bf18a
commit cd811cd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git 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
diff --git 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
@@ -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(