From afe5a282ac663508cb95f3e068fd0d0a72404b0f Mon Sep 17 00:00:00 2001
From: Matt Kornfield <kornfield@gretel.ai>
Date: Mon, 15 Jul 2024 09:29:13 -0700
Subject: [PATCH] Fix Failing CI tests

GitOrigin-RevId: 7a2ab75171d5240d78ef63af8bbbaaa572511295
---
 tests/gretel_client/integration/test_docker_agent.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tests/gretel_client/integration/test_docker_agent.py b/tests/gretel_client/integration/test_docker_agent.py
index 04090bf..8f6ae9e 100644
--- 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}")