fixed typo

devanshkv · May 6, 2020 · 4308e9b · 4308e9b
1 parent 258aa70
commit 4308e9b
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/bin/predict.py b/bin/predict.py
@@ -33,7 +33,7 @@
         logging.basicConfig(level=logging.INFO, format=logging_format)
 
     if args.model not in list(string.ascii_lowercase)[:11]:
-        raise ValueError(f'Model only range from a -- j.')
+        raise ValueError(f'Model only range from a -- k.')
 
     if args.gpu_id >= 0:
         os.environ["CUDA_VISIBLE_DEVICES"] = f'{args.gpu_id}'
@@ -46,9 +46,6 @@
     else:
         use_multiprocessing = False
 
-    if args.model not in list(string.ascii_lowercase)[:11]:
-        raise ValueError(f'Model only range from a -- j.')
-
     cands_to_eval = glob.glob(f'{args.data_dir}/*h5')
 
     if len(cands_to_eval) == 0: