Fix integration tests2 (#712)

Co-authored-by: Ubuntu <[email protected]>
predibase · Dec 10, 2024 · 37123f4 · 37123f4
1 parent 9a63db1
commit 37123f4
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/integration-tests/pytest.ini b/integration-tests/pytest.ini
@@ -0,0 +1,3 @@
+[pytest]
+log_cli=true
+log_level=DEBUG
diff --git a/integration-tests/requirements.txt b/integration-tests/requirements.txt
@@ -1,3 +1,3 @@
 requests==2.32.3
-docker==5.0.3
-pytest==7.4.0
+docker==7.1.0
+pytest==7.4.0
diff --git a/integration-tests/test_classifications.py b/integration-tests/test_classifications.py
@@ -19,5 +19,5 @@ def test_distilbert_ner():
             json={"inputs": "Johnny supports the Golden State Warriors. He lives in London."},
         )
         response.raise_for_status()
-        print("RESPONSE FROM CLASSIFICATION: ", response.json())
+        print("RESPONSE FROM CLASSIFICATION:", response.json())
         assert len(response.json()) > 0
diff --git a/integration-tests/utils/docker_runner.py b/integration-tests/utils/docker_runner.py
@@ -10,10 +10,6 @@
 logger = logging.getLogger(__name__)
 
 
-# def _init_client():
-#     return docker.from_env()
-
-
 def _init_client():
     try:
         client = docker.DockerClient(base_url="unix:///var/run/docker.sock", version="auto")