FIX-modin-project#6791: Pass additional environment variables to MPI …

…workers Signed-off-by: Igoshev, Iaroslav <[email protected]>
YarShev · Dec 1, 2023 · 14726f1 · 14726f1
1 parent 58a74fd
commit 14726f1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/modin/core/execution/unidist/common/utils.py b/modin/core/execution/unidist/common/utils.py
@@ -45,8 +45,9 @@ def initialize_unidist():
             unidist.init()
             """,
         )
-        # TODO: allow unidist to inherit env variables on initialization
-        # with set_env(PYTHONWARNINGS="ignore::FutureWarning"):
+        unidist_cfg.MpiRuntimeEnv.put(
+            {"env_vars": {"PYTHONWARNINGS": "ignore::FutureWarning"}}
+        )
         unidist.init()
 
     num_cpus = sum(v["CPU"] for v in unidist.cluster_resources().values())