Fixing num_threads

devin-petersohn · Sep 22, 2018 · ba3dd32 · ba3dd32
1 parent 49ccb05
commit ba3dd32
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/modin/pandas/__init__.py b/modin/pandas/__init__.py
@@ -22,6 +22,9 @@
 from .datetimes import to_datetime  # noqa: 402
 from .reshape import get_dummies # noqa: 402
 
+# Set this so that Pandas doesn't try to multithread by itself
+os.environ['OMP_NUM_THREADS'] = "1"
+
 try:
     if threading.current_thread().name == "MainThread":
         ray.init(
@@ -31,9 +34,6 @@
 except AssertionError:
     pass
 
-# Set this so that Pandas doesn't try to multithread by itself
-os.environ['OMP_NUM_THREADS'] = "1"
-
 num_cpus = ray.global_state.cluster_resources()['CPU']
 DEFAULT_NPARTITIONS = int(num_cpus)