Fix Failing CI tests

GitOrigin-RevId: 7a2ab75171d5240d78ef63af8bbbaaa572511295
gretelai · Jul 15, 2024 · afe5a28 · afe5a28
1 parent e26639e
commit afe5a28
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/gretel_client/integration/test_docker_agent.py b/tests/gretel_client/integration/test_docker_agent.py
@@ -45,7 +45,14 @@ def test_docker_agent(agent_config: AgentConfig, request):
     model = project.create_model_obj(model_config, fake_pii)
 
     model.submit_manual()
-    request.addfinalizer(model.cancel)
+
+    def model_cancel_ignore_exception():
+        try:
+            model.cancel()
+        except Exception as e:
+            print(e)
+
+    request.addfinalizer(model_cancel_ignore_exception)
 
     print(f"launched model {model.id}")