X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=cipher%2Fcolumn_transposition.py;fp=cipher%2Fcolumn_transposition.py;h=7e0fc28a3d002208ccfd0ea0c6958216aa61fa09;hb=df271527a6e3e4d10da9f9be3d1e9ebea9171493;hp=a141ff2c1f8fcc263a43884f307bfdb3f8003d8a;hpb=311b300d197536622980f7a837294d8245e326b4;p=cipher-tools.git

diff --git a/cipher/column_transposition.py b/cipher/column_transposition.py
index a141ff2..7e0fc28 100644
--- 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))]