Merge pull request #320 from plasma-umass/shim_get_context

Added in shim for get_context

Author: sternj

scalene/replacement_get_context.py

@@ -0,0 +1,10 @@
+from scalene.scalene_profiler import Scalene
+import multiprocessing
+from typing import Any
+
+@Scalene.shim
+def replacement_mp_get_context(scalene: Scalene) -> None:
+    old_get_context = multiprocessing.get_context
+    def replacement_get_context(method: Any):
+        return old_get_context('fork')
+    multiprocessing.get_context = replacement_get_context

scalene/scalene_profiler.py

@@ -472,6 +472,7 @@ def __init__(
         import scalene.replacement_mp_lock
         import scalene.replacement_pjoin
         import scalene.replacement_thread_join
+        import scalene.replacement_get_context
 
         if sys.platform != "win32":
             import scalene.replacement_fork