data-manipulation-utilities 0.2.8.dev733__py3-none-any.whl → 0.2.8.dev736__py3-none-any.whl

{data_manipulation_utilities-0.2.8.dev733.dist-info → data_manipulation_utilities-0.2.8.dev736.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: data-manipulation-utilities
-Version: 0.2.8.dev733
+Version: 0.2.8.dev736
 Summary: Project storing utilities needed to reduce boilerplate code when analyzing data
 Requires-Python: >=3.10,<3.13
 Classifier: Programming Language :: Python :: 3

{data_manipulation_utilities-0.2.8.dev733.dist-info → data_manipulation_utilities-0.2.8.dev736.dist-info}/RECORD

@@ -39,7 +39,7 @@ dmu/stats/zfit_plotter.py,sha256=gbN5KxhJcP4ItCi98c-fj5_UtvVWL_NA9jkTHiRjvnE,238
 dmu/testing/utilities.py,sha256=WYlz7Ve5lQjuWhhNL4gWe6_qcByBLV762Lhrc6A0P9E,7421
 dmu/text/transformer.py,sha256=4lrGknbAWRm0-rxbvgzOO-eR1-9bkYk61boJUEV3cQ0,6100
 dmu/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dmu/workflow/cache.py,sha256=CtkGwxuF4UJlD55SmUJcRgWYLsbZOyUvYLI8oTVzk_g,8768
+dmu/workflow/cache.py,sha256=so9BZt03vxNGHKcDPYpzz4T-CsLPUMmj_3EO3HXpLIc,8947
 dmu_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dmu_data/ml/tests/diagnostics_from_file.yaml,sha256=quvXOPkRducnBsctyape_Rn5_aqMEpPo6nO_UweMORo,404
 dmu_data/ml/tests/diagnostics_from_model.yaml,sha256=rtCQlmGS9ld2xoQJEE35nA07yfRMklEfQEW0w3gRv2A,261
@@ -77,7 +77,7 @@ dmu_data/text/transform.txt,sha256=EX760da6Vkf-_EPxnQlC5hGSkfFhJCCGCD19NU-1Qto,4
 dmu_data/text/transform_set.toml,sha256=Jeh7BTz82idqvbOQJtl9-ur56mZkzDn5WtvmIb48LoE,150
 dmu_data/text/transform_set.txt,sha256=1KivMoP9LxPn9955QrRmOzjEqduEjhTetQ9MXykO5LY,46
 dmu_data/text/transform_trf.txt,sha256=zxBRTgcSmX7RdqfmWF88W1YqbyNHa4Ccruf1MmnYv2A,74
-data_manipulation_utilities-0.2.8.dev733.dist-info/METADATA,sha256=v5qmztFZBzoriqplP_EvZ5nhowvDKXS7dchJQ6QNgHQ,51716
-data_manipulation_utilities-0.2.8.dev733.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-data_manipulation_utilities-0.2.8.dev733.dist-info/entry_points.txt,sha256=M0C8_u9B_xSmyfemdPwdIBh9QuPIkjhEpG060Y5_Pjw,321
-data_manipulation_utilities-0.2.8.dev733.dist-info/RECORD,,
+data_manipulation_utilities-0.2.8.dev736.dist-info/METADATA,sha256=TmvJPwuZT1s9sA0IVrkUpNhQaTjzj9rFEm-cTan915U,51716
+data_manipulation_utilities-0.2.8.dev736.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+data_manipulation_utilities-0.2.8.dev736.dist-info/entry_points.txt,sha256=M0C8_u9B_xSmyfemdPwdIBh9QuPIkjhEpG060Y5_Pjw,321
+data_manipulation_utilities-0.2.8.dev736.dist-info/RECORD,,

dmu/workflow/cache.py

@@ -63,19 +63,6 @@ class Cache:
         self._cache_dir = self._get_dir(kind='cache')
         self._hash_dir  : str
     # ---------------------------
-    @classmethod
-    def set_cache_root(cls, root : str) -> None:
-        '''
-        Sets the path to the directory WRT which the _out_path_
-        will be placed
-        '''
-        if cls._cache_root is not None:
-            raise ValueError(f'Trying to set {root}, but already found {cls._cache_root}')
-
-        os.makedirs(root, exist_ok=True)
-
-        cls._cache_root = root
-    # ---------------------------
     def _get_code_hash(self) -> str:
         '''
         If `MyTool` inherits from `Cache`. `mytool.py` git commit hash
@@ -242,9 +229,8 @@ class Cache:
 
         return True
     # ---------------------------
-    @contextmanager
-    @staticmethod
-    def turn_off_cache(val : list[str]|None):
+    @classmethod
+    def turn_off_cache(cls, val):
         '''
         Parameters
         ------------------
@@ -256,11 +242,30 @@ class Cache:
             log.error('This manager expects: list[str]|None')
             raise ValueError(f'Invalid value: {val}')
 
-        old_val = Cache._l_skip_class
+        old_val = cls._l_skip_class
+        @contextmanager
+        def _context():
+            cls._l_skip_class = val
+            try:
+                yield
+            finally:
+                cls._l_skip_class = old_val
 
-        Cache._l_skip_class = val
-        try:
-            yield
-        finally:
-            Cache._l_skip_class = old_val
+        return _context()
+    # ---------------------------
+    @classmethod
+    def set_cache_root(cls, root : str) -> None:
+        '''
+        Sets the path to the directory WRT which the _out_path_
+        will be placed.
+
+        This is meant to be called once per execution and has a lock
+        that will raise an exception if called twice
+        '''
+        if cls._cache_root is not None:
+            raise ValueError(f'Trying to set {root}, but already found {cls._cache_root}')
+
+        os.makedirs(root, exist_ok=True)
+
+        cls._cache_root = root
 # ---------------------------