data-manipulation-utilities 0.2.8.dev733__tar.gz → 0.2.8.dev736__tar.gz

{data_manipulation_utilities-0.2.8.dev733 → data_manipulation_utilities-0.2.8.dev736}/PKG-INFO

@@ -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 → data_manipulation_utilities-0.2.8.dev736}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name           = 'data-manipulation-utilities'
 description    = 'Project storing utilities needed to reduce boilerplate code when analyzing data'
-version        = "0.2.8.dev733"
+version        = "0.2.8.dev736"
 readme         = 'README.md'
 package-mode   = true 
 packages       = [

{data_manipulation_utilities-0.2.8.dev733 → data_manipulation_utilities-0.2.8.dev736}/src/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
 # ---------------------------