push

hassiebp · hassiebp · commit a93acca7fed5 · 2025-08-05T16:56:03.000+02:00
diff --git a/tests/test_decorators.py b/tests/test_decorators.py
@@ -24,9 +24,10 @@
 
 
 def removeMockResourceManagerInstances():
-    for public_key in list(LangfuseResourceManager._instances.keys()):
-        if public_key != "pk-lf-1234567890":
-            LangfuseResourceManager._instances.pop(public_key)
+    with LangfuseResourceManager._lock:
+        for public_key in list(LangfuseResourceManager._instances.keys()):
+            if public_key != os.getenv(LANGFUSE_PUBLIC_KEY):
+                LangfuseResourceManager._instances.pop(public_key)
 
 
 def test_nested_observations():
