logging

predibase · Nov 8, 2024 · faabd0b · faabd0b
1 parent 24576f7
commit faabd0b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/integration-tests-2/test_base_llms.py b/integration-tests-2/test_base_llms.py
@@ -7,11 +7,16 @@
 def run_lorax_container(config):
     runner = DockerModelRunner()
     runner.start_container(config)
-    runner.wait_for_healthy()
+    print("Started container")
     try:
+        print("Waiting for container to be healthy")
+        runner.wait_for_healthy()
+        print("Container is healthy")
         yield
     finally:
+        print("Stopping container")
         runner.stop_container()
+        print("Container stopped")
 
 
 def test_base_mistral():