Fix cachedir type

scripts/train.py

@@ -40,7 +40,7 @@
 from trackastra.utils import (
     blockwise_causal_norm,
     blockwise_sum,
-    none_or_path,
+    none_or_str,
     normalize,
     preallocate_memory,
     random_label_cmap,
@@ -1087,7 +1087,7 @@ def parse_train_args():
     )
     parser.add_argument(
         "--cachedir",
-        type=none_or_path,
+        type=none_or_str,
         default=".cache",
         help="cache dir for CTCData. Set to `None` to disable caching.",
     )

trackastra/utils/__init__.py

@@ -4,6 +4,7 @@
     blockwise_causal_norm,
     blockwise_sum,
     none_or_path,
+    none_or_str,
     normalize,
     preallocate_memory,
     random_label_cmap,

trackastra/utils/utils.py

@@ -464,6 +464,14 @@ def str2path(x: str) -> Path:
         return Path(x).expanduser().resolve()
 
 
+def none_or_str(x: str | None) -> str | None:
+    """Allow for none value in string parsing."""
+    if x in (None, "None", "none"):
+        return None
+    else:
+        return str(x)
+
+
 def none_or_path(x: str | None) -> Path | None:
     """Cast string to resolved absolute path or return None."""
     if x in (None, "None", "none"):