From 8e5d0871c89ac5d87d51ecab9309178276ee1943 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 11 Mar 2023 08:06:22 -0600 Subject: [PATCH] log which cache items have been pruned --- api/onnx_web/server/model_cache.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/server/model_cache.py b/api/onnx_web/server/model_cache.py index d757a94f..7f1ef522 100644 --- a/api/onnx_web/server/model_cache.py +++ b/api/onnx_web/server/model_cache.py @@ -47,9 +47,14 @@ class ModelCache: def prune(self): total = len(self.cache) - if total > self.limit: + overage = total - self.limit + if overage > 0: + removed = self.cache[:overage] logger.info( - "removing models from cache, %s of %s", (total - self.limit), total + "removing %s of %s models from cache, %s", + overage, + total, + [m[0] for m in removed], ) self.cache[:] = self.cache[-self.limit :] else: