diff --git a/lint.sh b/lint.sh index 1679b2381979a9aca4934362f9a3f43c2d597fb3..38b7859bf494fef4625d1b26649ea7bce0ea5caf 100755 --- a/lint.sh +++ b/lint.sh @@ -10,6 +10,6 @@ set -o errexit find . -name '*.py' -not -path "./docs/*" \ | xargs pylint \ --errors-only \ - --disable=print-statement,unsubscriptable-object,not-an-iterable,no-member + --disable=unsubscriptable-object,not-an-iterable,no-member echo 'Passes pylint check' diff --git a/mhcflurry/parallelism.py b/mhcflurry/parallelism.py index 89c42cf3f5749d249fdfa4589697d8825c0be61c..b8ed6466781d95791887db25d91260521c4d95da 100644 --- a/mhcflurry/parallelism.py +++ b/mhcflurry/parallelism.py @@ -1,5 +1,5 @@ from multiprocessing import Pool, Queue, cpu_count -from six.moves.queue import Empty +from six.moves import queue from multiprocessing.util import Finalize from pprint import pprint @@ -86,7 +86,7 @@ def worker_init_entry_point( if arg_queue: try: kwargs = arg_queue.get(block=False) - except Empty: + except queue.Empty: print("Argument queue empty. Using round robin arg queue.") kwargs = backup_arg_queue.get(block=True) backup_arg_queue.put(kwargs)