Merge pull request #485 from ChristopherBradley/universe

BUG: reordered the warning.warn arguments in parallel.py, fixes #484
cogent3 · Jan 10, 2020 · cb50a62 · cb50a62
2 parents d121884 + 4f9f749
commit cb50a62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cogent3/util/parallel.py b/src/cogent3/util/parallel.py
@@ -139,13 +139,13 @@ def imap(f, s, max_workers=None, use_mpi=False, if_serial="raise", chunksize=Non
         if COMM.Get_attr(MPI.UNIVERSE_SIZE) == 1 and if_serial == "raise":
             raise RuntimeError(err_msg)
         elif COMM.Get_attr(MPI.UNIVERSE_SIZE) == 1 and if_serial == "warn":
-            warnings.warn(UserWarning, msg=err_msg)
+            warnings.warn(err_msg, UserWarning)
 
         max_workers = max_workers or 0
 
         if max_workers > COMM.Get_attr(MPI.UNIVERSE_SIZE):
             warnings.warn(
-                UserWarning, msg="max_workers too large, reducing to UNIVERSE_SIZE-1"
+                "max_workers too large, reducing to UNIVERSE_SIZE-1", UserWarning
             )
 
         max_workers = min(max_workers, COMM.Get_attr(MPI.UNIVERSE_SIZE) - 1)