Make cache version format less restrictive

AllenInstitute · Jan 10, 2024 · 616aaa9 · 616aaa9
1 parent 55264e5
commit 616aaa9
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/npc_lims/paths/cache.py b/src/npc_lims/paths/cache.py
@@ -66,16 +66,23 @@ def get_current_cache_version() -> str:
 
 
 def _parse_version(version: str) -> str:
-    return f"v{packaging.version.parse(str(version))}"
+    try:
+        return f"v{packaging.version.parse(str(version))}"
+    except packaging.version.InvalidVersion:
+        raise ValueError(f"Invalid version {version!r}")
 
 
 def _parse_cache_path(
     nwb_component: NWBComponentStr,
     session_id: str | npc_session.SessionRecord | None = None,
     version: str | None = None,
 ) -> upath.UPath:
-    version = _parse_version(version) if version else get_current_cache_version()
-    d = CACHE_ROOT / version / nwb_component
+    version = version or get_current_cache_version()
+    d = (
+        CACHE_ROOT
+        / version
+        / nwb_component
+    )
     if session_id is None:
         return d
     return (