Add set_cache_tll

Author: p-holl

phiml/dataclasses/__init__.py

@@ -47,6 +47,6 @@
 from ._dataclasses import sliceable, data_fields, non_data_fields, config_fields, special_fields, replace, copy, getitem, equal, data_eq
 
 from ._parallel import parallel_compute, parallel_property
-from ._tensor_cache import on_load_into_memory, get_cache_files
+from ._tensor_cache import on_load_into_memory, get_cache_files, set_cache_tll
 
 __all__ = [key for key in globals().keys() if not key.startswith('_')]

phiml/dataclasses/_tensor_cache.py

@@ -250,3 +250,11 @@ def cleaner():
 
 
 _CACHE = SoftCache(2., [])
+
+
+def set_cache_tll(ttl_seconds: float):
+    """
+    Sets the time to live (TTL) for data loaded into memory for disk-backed tensors.
+    This function should be called before the tensor cache is used.
+    """
+    _CACHE.ttl = ttl_seconds