reformat code using black

Netflix · Jun 22, 2023 · 8c81441 · 8c81441
1 parent bea58fd
commit 8c81441
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 12 deletions.
diff --git a/metaflow/multicore_utils.py b/metaflow/multicore_utils.py
@@ -5,6 +5,7 @@
 from multiprocessing import cpu_count
 from tempfile import NamedTemporaryFile
 import metaflow.tracing as tracing
+
 try:
     # Python 2
     import cPickle as pickle
@@ -60,15 +61,13 @@ def _spawn(func, arg, dir):
 
 
 def parallel_imap_unordered(func, iterable, max_parallel=None, dir=None):
-
     if max_parallel is None:
         max_parallel = cpu_count()
 
     args_iter = iter(iterable)
     pids = [_spawn(func, arg, dir) for arg in islice(args_iter, max_parallel)]
 
     while pids:
-
         pid, output_file = pids.pop()
         if os.waitpid(pid, 0)[1]:
             raise MulticoreException("Child failed")

diff --git a/metaflow/sidecar/sidecar_subprocess.py b/metaflow/sidecar/sidecar_subprocess.py
@@ -68,7 +68,6 @@ def __init__(self, worker_type):
         self.start()
 
     def start(self):
-
         if (
             self._worker_type is not None
             and self._worker_type.startswith(NULL_SIDECAR_PREFIX)

diff --git a/metaflow/tracing/__init__.py b/metaflow/tracing/__init__.py
@@ -49,6 +49,7 @@ def wrapper_func(*args, **kwargs):
 
     return wrapper_func
 
+
 if CONSOLE_TRACE_ENABLED or OTEL_ENDPOINT or ZIPKIN_ENDPOINT:
     try:
         from .tracing_modules import *

diff --git a/metaflow/tracing/span_exporter.py b/metaflow/tracing/span_exporter.py
@@ -2,20 +2,21 @@
 from metaflow.metaflow_config import (
     OTEL_ENDPOINT,
     ZIPKIN_ENDPOINT,
-    CONSOLE_TRACE_ENABLED
+    CONSOLE_TRACE_ENABLED,
 )
 
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
     OTLPSpanExporter,
 )
 from opentelemetry.exporter.zipkin.proto.http import ZipkinExporter
 
-from opentelemetry.sdk.trace.export import  ConsoleSpanExporter
+from opentelemetry.sdk.trace.export import ConsoleSpanExporter
+
 
 def get_span_exporter():
     if OTEL_ENDPOINT:
         return set_otel_exporter()
-                
+
     elif ZIPKIN_ENDPOINT:
         return set_zipkin_exporter()
     elif CONSOLE_TRACE_ENABLED:
@@ -25,7 +26,6 @@ def get_span_exporter():
         return
 
 
-
 def set_otel_exporter():
     from metaflow.metaflow_config import (
         SERVICE_AUTH_KEY,
@@ -49,13 +49,14 @@ def set_otel_exporter():
         return
     return span_exporter
 
+
 def set_zipkin_exporter():
     span_exporter = ZipkinExporter(
-                endpoint=ZIPKIN_ENDPOINT,
-            )
+        endpoint=ZIPKIN_ENDPOINT,
+    )
     return span_exporter
 
+
 def set_console_exporter():
     span_exporter = ConsoleSpanExporter()
     return span_exporter
-
diff --git a/metaflow/tracing/tracing_modules.py b/metaflow/tracing/tracing_modules.py
@@ -15,6 +15,7 @@
 
 tracer_provider = None
 
+
 def init_tracing():
     global tracer_provider
     if tracer_provider is not None:
@@ -26,7 +27,6 @@ def init_tracing():
     set_global_textmap(EnvPropagator(None))
     span_exporter = get_span_exporter()
 
-
     if "METAFLOW_KUBERNETES_POD_NAMESPACE" in os.environ:
         service_name = "metaflow-kubernetes"
     elif "AWS_BATCH_JOB_ID" in os.environ:
@@ -47,7 +47,6 @@ def init_tracing():
     RequestsInstrumentor().instrument()
 
 
-
 @contextlib.contextmanager
 def post_fork():
     global tracer_provider