revert tests

basetenlabs · Feb 27, 2025 · 7f957db · 7f957db
1 parent a0ba879
commit 7f957db
Showing 1 changed file with 15 additions and 18 deletions.
diff --git a/truss/tests/test_model_inference.py b/truss/tests/test_model_inference.py
@@ -957,14 +957,14 @@ def predict(self, model_input):
         restart_threshold_seconds: 1700
     """
 
-    # with ensure_kill_all(), _temp_truss(model, config) as tr:
-    #     _ = tr.docker_run(local_port=8090, detach=True, wait_for_server_ready=True)
+    with ensure_kill_all(), _temp_truss(model, config) as tr:
+        _ = tr.docker_run(local_port=8090, detach=True, wait_for_server_ready=True)
 
-    #     assert tr.spec.config.runtime.health_checks.restart_check_delay_seconds == 100
-    #     assert tr.spec.config.runtime.health_checks.restart_threshold_seconds == 1700
-    #     assert (
-    #         tr.spec.config.runtime.health_checks.stop_traffic_threshold_seconds is None
-    #     )
+        assert tr.spec.config.runtime.health_checks.restart_check_delay_seconds == 100
+        assert tr.spec.config.runtime.health_checks.restart_threshold_seconds == 1700
+        assert (
+            tr.spec.config.runtime.health_checks.stop_traffic_threshold_seconds is None
+        )
 
     config = """runtime:
     health_checks:
@@ -974,23 +974,20 @@ def predict(self, model_input):
     """
 
     with ensure_kill_all(), _temp_truss(model, config) as tr:
-        container = tr.docker_run(
-            local_port=8090, detach=True, wait_for_server_ready=True
-        )
+        _ = tr.docker_run(local_port=8090, detach=True, wait_for_server_ready=True)
 
         assert tr.spec.config.runtime.health_checks.restart_check_delay_seconds == 1200
         assert tr.spec.config.runtime.health_checks.restart_threshold_seconds == 90
         assert tr.spec.config.runtime.health_checks.stop_traffic_threshold_seconds == 50
-        print(container.logs())
 
-    # with ensure_kill_all(), _temp_truss(model, "") as tr:
-    #     _ = tr.docker_run(local_port=8090, detach=True, wait_for_server_ready=True)
+    with ensure_kill_all(), _temp_truss(model, "") as tr:
+        _ = tr.docker_run(local_port=8090, detach=True, wait_for_server_ready=True)
 
-    #     assert tr.spec.config.runtime.health_checks.restart_check_delay_seconds is None
-    #     assert tr.spec.config.runtime.health_checks.restart_threshold_seconds is None
-    #     assert (
-    #         tr.spec.config.runtime.health_checks.stop_traffic_threshold_seconds is None
-    #     )
+        assert tr.spec.config.runtime.health_checks.restart_check_delay_seconds is None
+        assert tr.spec.config.runtime.health_checks.restart_threshold_seconds is None
+        assert (
+            tr.spec.config.runtime.health_checks.stop_traffic_threshold_seconds is None
+        )
 
 
 @pytest.mark.integration