All tests now running and passing

[cipher-tools.git] / cipher / column_transposition.py

diff --git a/cipher/column_transposition.py b/cipher/column_transposition.py
index a141ff2c1f8fcc263a43884f307bfdb3f8003d8a..7e0fc28a3d002208ccfd0ea0c6958216aa61fa09 100644 (file)
--- a/cipher/column_transposition.py
+++ b/cipher/column_transposition.py
@@ -1,7 +1,8 @@
-from utilities import *
-from language_models import *
+import math
 import multiprocessing 
 from itertools import chain
+from support.utilities import *
+from support.language_models import *
 
 from logger import logger
 
@@ -254,7 +255,7 @@ def scytale_break_mp(message, max_key_length=20,
         fitness=Ptrigrams) # doctest: +ELLIPSIS
     (5, -997.0129085...)
     """
-    with Pool() as pool:
+    with multiprocessing.Pool() as pool:
         helper_args = [(message, trans, False, True, fitness)
             for trans in
                 [[col for col in range(math.ceil(len(message)/rows))]