feat(python): don't raise error but print a warning if mp fork method… (

#5342)
pola-rs · Oct 26, 2022 · 1b0c5a8 · 1b0c5a8
1 parent e76d5d9
commit 1b0c5a8
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/py-polars/polars/mp_check.py b/py-polars/polars/mp_check.py
@@ -39,13 +39,15 @@ def mp_check() -> None:
         #   - "fork": Still the default on Linux.
         mp.set_start_method("spawn")
     elif mp_method == "fork":
-        raise ImportError(
-            "Polars only works with python multiprocessing method set to: "
-            '`mp.set_start_method("spawn")` or `mp.set_start_method("forkserver")`'
-            "before importing polars. "
-            "See: https://docs.python.org/3/library/multiprocessing.html"
-            "#contexts-and-start-methods"
-        )
+        import warnings
+
+        msg = "Polars only works with python multiprocessing method set to: "
+        '`mp.set_start_method("spawn")` or `mp.set_start_method("forkserver")`'
+        "before importing polars. "
+        "See: https://docs.python.org/3/library/multiprocessing.html"
+        "#contexts-and-start-methods"
+
+        warnings.warn(msg, ImportWarning)
 
 
 mp_check()