More cleanup, yey!

[python_utils.git] / parallelize.py

diff --git a/parallelize.py b/parallelize.py
index 98f883c5716ae8fa3d67101b1d38d20dece7a0b6..b2a1cedb988d98d78cf7654e3f65d4dfaa8beff6 100644 (file)
--- a/parallelize.py
+++ b/parallelize.py
@@ -2,22 +2,23 @@
 
 """A decorator to help with dead simple parallelization."""
 
+
 import atexit
-from enum import Enum
 import functools
 import typing
+from enum import Enum
 
 
 class Method(Enum):
+    """How should we parallelize; by threads, processes or remote workers?"""
+
     THREAD = 1
     PROCESS = 2
     REMOTE = 3
 
 
 def parallelize(
-    _funct: typing.Optional[typing.Callable] = None,
-    *,
-    method: Method = Method.THREAD
+    _funct: typing.Optional[typing.Callable] = None, *, method: Method = Method.THREAD
 ) -> typing.Callable:
     """Usage: