explicitly import tracing functions from tracing_modules

Netflix · Oct 18, 2023 · 85a43dc · 85a43dc
1 parent 9c31d3b
commit 85a43dc
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/metaflow/tracing/__init__.py b/metaflow/tracing/__init__.py
@@ -54,7 +54,16 @@ def wrapper_func(*args, **kwargs):
     CONSOLE_TRACE_ENABLED or OTEL_ENDPOINT or ZIPKIN_ENDPOINT
 ):
     try:
-        from .tracing_modules import *
+        # Overrides No-Op implementations if a specific provider is configured.
+        from .tracing_modules import (
+            init_tracing,
+            post_fork,
+            cli_entrypoint,
+            inject_tracing_vars,
+            get_trace_id,
+            traced,
+            tracing,
+        )
 
     except ImportError as e:
         print(e.msg)