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