From a7158f6ec72d5dd755a7e757409e42be17212693 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 2 Sep 2019 11:09:25 -0400
Subject: [PATCH] fix
---
mhcflurry/class1_neural_network.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 1ebf7230..44f34d4f 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -20,9 +20,9 @@ from .data_dependent_weights_initialization import lsuv_init
DEFAULT_PREDICT_BATCH_SIZE = 4096
if os.environ.get("MHCFLURRY_DEFAULT_PREDICT_BATCH_SIZE"):
- DEFAULT_PREDICT_BATCH_SIZE = os.environ[
+ DEFAULT_PREDICT_BATCH_SIZE = int(os.environ[
"MHCFLURRY_DEFAULT_PREDICT_BATCH_SIZE"
- ]
+ ])
logging.info(
"Configured default predict batch size: %d" % DEFAULT_PREDICT_BATCH_SIZE)
--
GitLab