diff --git a/metaflow/tracing/__init__.py b/metaflow/tracing/__init__.py index aadb7793c6c..5472c71258c 100644 --- 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)