fix

predibase · Nov 11, 2024 · fdb763c · fdb763c
1 parent 2ae543a
commit fdb763c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/integration-tests/action.yaml b/.github/workflows/integration-tests/action.yaml
@@ -56,10 +56,10 @@ runs:
       shell: bash
       run: |
         cd integration-tests
-        pytest tests/test_embeddings.py -vv --capture=tee-sys --log-cli-level=INFO
+        pytest test_embeddings.py -vv --capture=tee-sys --log-cli-level=INFO
 
     - name: Run LLM tests
       shell: bash
       run: |
         cd integration-tests
-        pytest tests/test_base_llms.py -vv --capture=tee-sys --log-cli-level=INFO
+        pytest test_base_llms.py -vv --capture=tee-sys --log-cli-level=INFO
diff --git a/integration-tests/tests/test_base_llms.py → integration-tests/test_base_llms.py b/integration-tests/tests/test_base_llms.py → integration-tests/test_base_llms.py
@@ -1,5 +1,5 @@
 import requests
-from ..utils.docker_runner import run_lorax_container
+from .utils.docker_runner import run_lorax_container
 
 
 def test_base_mistral():

diff --git a/integration-tests/tests/test_embeddings.py → integration-tests/test_embeddings.py b/integration-tests/tests/test_embeddings.py → integration-tests/test_embeddings.py
@@ -1,5 +1,5 @@
 import requests
-from ..utils.docker_runner import run_lorax_container
+from .utils.docker_runner import run_lorax_container
 
 
 def test_stella_1_5b():