fix decorator returntype

OpenMined · Sep 26, 2024 · 01d77f2 · 01d77f2
1 parent bd72687
commit 01d77f2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/syft/src/syft/util/telemetry.py b/packages/syft/src/syft/util/telemetry.py
@@ -3,6 +3,7 @@
 import logging
 import os
 from typing import Any
+from typing import TYPE_CHECKING
 from typing import TypeVar
 
 # relative
@@ -131,4 +132,8 @@ def instrument_sqlalchemny() -> None:
         logger.error(f"Failed to load SQLAlchemyInstrumentor. {e}")
 
 
-instrument = setup_instrumenter()
+if TYPE_CHECKING:
+    # To let static type checker know the returntype of instrument decorators
+    instrument = no_instrument
+else:
+    instrument = setup_instrumenter()