Made local cache version dependent.

hendriks73 · Oct 11, 2024 · 6bc8daa · 6bc8daa
1 parent 3a5bde8
commit 6bc8daa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tempocnn/classifier.py b/tempocnn/classifier.py
@@ -7,6 +7,7 @@
 
 import numpy as np
 from tensorflow.keras.models import load_model
+from tempocnn.version import __version__ as package_version
 
 logger = logging.getLogger("tempocnn.classifier")
 
@@ -300,7 +301,7 @@ def _to_model_resource(model_name):
 
 def _extract_from_package(resource):
     # check local cache
-    cache_path = Path(Path.home(), ".tempocnn", resource)
+    cache_path = Path(Path.home(), ".tempocnn", package_version, resource)
     if cache_path.exists():
         return str(cache_path)