From 8dc1ea7f1f057d07ccb20bed095db33a8f64eb1d Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 13 Feb 2023 18:15:02 -0600 Subject: [PATCH] declare missing cache --- api/onnx_web/server/model_cache.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/server/model_cache.py b/api/onnx_web/server/model_cache.py index 4e80f377..8e171874 100644 --- a/api/onnx_web/server/model_cache.py +++ b/api/onnx_web/server/model_cache.py @@ -9,10 +9,11 @@ class ModelCache: limit: int def __init__(self, limit: int) -> None: + self.cache = [] self.limit = limit def drop(self, tag: str, key: Any) -> None: - self.cache = [ + self.cache[:] = [ model for model in self.cache if model[0] != tag and model[1] != key ]