From 75a3278f458a8e580042636d79a5098a34c6de43 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 9 Sep 2019 13:08:25 -0400
Subject: [PATCH] fix
---
mhcflurry/testing_utils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mhcflurry/testing_utils.py b/mhcflurry/testing_utils.py
index 8c14fa15..f6cc079b 100644
--- a/mhcflurry/testing_utils.py
+++ b/mhcflurry/testing_utils.py
@@ -3,5 +3,5 @@ from . import Class1NeuralNetwork
def module_cleanup():
import keras.backend as K
- Class1NeuralNetwork.KERAS_MODELS_CACHE.clear()
+ Class1NeuralNetwork.clear_model_cache()
K.clear_session()
--
GitLab