Merge pull request #18 from mskcc/hotfix/ensure_coverage_is_numeric

Fix coverage param
mskcc · Nov 16, 2020 · c9d8dd8 · c9d8dd8
2 parents cb4395c + 1c1c343
commit c9d8dd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main/evaluate_sample.py b/main/evaluate_sample.py
@@ -174,7 +174,7 @@ def run_eval(
     print("Evaluating Samples, Lets go!!!")
     loader_kwargs = {"num_workers": 1, "pin_memory": True} if cuda else {}
     eval_loader = data_utils.DataLoader(
-        MSIBags(vector_location, coverage, False, False),
+        MSIBags(vector_location, int(coverage), False, False),
         batch_size=1,
         shuffle=False,
         **loader_kwargs