fix

Author: hassiebp

langfuse/_client/resource_manager.py

@@ -256,7 +256,8 @@ def _initialize_instance(
 
     @classmethod
     def reset(cls):
-        cls._instances.clear()
+        for key in cls._instances:
+            cls._instances.pop(key).shutdown()
 
     def add_score_task(self, event: dict, *, force_sample: bool = False):
         try:

tests/test_core_sdk.py

@@ -7,6 +7,7 @@
 import pytest
 
 from langfuse import Langfuse
+from langfuse._client.resource_manager import LangfuseResourceManager
 from langfuse._utils import _get_timestamp
 from tests.api_wrapper import LangfuseAPI
 from tests.utils import (
@@ -1781,6 +1782,8 @@ def test_create_trace_sampling_zero():
 
 
 def test_mask_function():
+    LangfuseResourceManager.reset()
+
     def mask_func(data):
         if isinstance(data, dict):
             if "should_raise" in data: