removed wrapper function

Lightning-AI · Aug 31, 2020 · 0cf6eaa · 0cf6eaa
1 parent 944db7f
commit 0cf6eaa
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/pytorch_lightning/utilities/xla_device_utils.py b/pytorch_lightning/utilities/xla_device_utils.py
@@ -19,14 +19,11 @@ def inner_f(queue, func, **kwargs):
 
 
 def pl_multi_process(func):
-    def wrapper(*args, **kwargs):
-        queue = Queue()
-        proc = Process(target=inner_f, args=(queue, func,), kwargs=kwargs)
-        proc.start()
-        proc.join()
-        return queue.get()
-
-    return wrapper
+    queue = Queue()
+    proc = Process(target=inner_f, args=(queue, func,))
+    proc.start()
+    proc.join()
+    return queue.get()
 
 
 def fetch_xla_device_type(device):