add cleanup to env integration tests

basetenlabs · Oct 22, 2024 · ba0e5fd · ba0e5fd
1 parent db748cf
commit ba0e5fd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/truss/tests/test_model_inference.py b/truss/tests/test_model_inference.py
@@ -798,6 +798,13 @@ def predict(self, model_input):
             in container.logs()
         )
         assert "in beta environment" in container.logs()
+        container.execute(
+            [
+                "bash",
+                "-c",
+                "rm -f /etc/b10_dynamic_config/environment",
+            ]
+        )
 
     # Test a truss that uses the environment in load()
     model = """