fix

mhcflurry/regression_target.py

@@ -14,7 +14,7 @@ def from_ic50(ic50, max_ic50=50000.0):
     numpy.array of float
 
     """
-    x = 1.0 - (numpy.log(ic50) / numpy.log(max_ic50))
+    x = 1.0 - (numpy.log(numpy.maximum(ic50, 1e-12)) / numpy.log(max_ic50))
     return numpy.minimum(
         1.0,
         numpy.maximum(0.0, x))

mhcflurry/train_pan_allele_models_command.py

@@ -336,6 +336,7 @@ def main(args):
     start = time.time()
 
     worker_pool = worker_pool_with_gpu_assignments_from_args(args)
+    print("Worker pool", worker_pool)
 
     if worker_pool:
         print("Processing %d work items in parallel." % len(work_items))

test/test_train_pan_allele_models_command.py

@@ -136,3 +136,10 @@ if os.environ.get("KERAS_BACKEND") != "theano":
 
 def test_run_serial():
     run_and_check(n_jobs=1)
+
+if __name__ == "__main__":
+    test_run_serial()
+    #for (name, value) in list(globals().items()):
+    #    if name.startswith("test_"):
+    #        print("Running test", name)
+    #        value()
\ No newline at end of file