fixup! replace response mock with pytest-httpx (mock)

cognitedata · Feb 6, 2025 · 622ca10 · 622ca10
1 parent bea8b5b
commit 622ca10
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/utils.py b/tests/utils.py
@@ -175,10 +175,14 @@ def random_gamma_dist_integer(inclusive_max, max_tries=100):
 
 @contextmanager
 def set_max_workers(cognite_client, new):
-    old = cognite_client._config.max_workers
-    cognite_client._config.max_workers = new
-    yield
-    cognite_client._config.max_workers = old
+    from cognite.client import global_config
+
+    old = global_config.max_workers
+    global_config.max_workers = new
+    try:
+        yield
+    finally:
+        global_config.max_workers = old
 
 
 @contextmanager