nshutils 0.31.0__py3-none-any.whl → 0.31.1__py3-none-any.whl

nshutils/actsave/_saver.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import contextlib
 import fnmatch
+import os
 import tempfile
 import weakref
 from collections.abc import Callable, Mapping
@@ -266,7 +267,11 @@ class ActSaveProvider:
 
         if save_dir is None:
             save_dir = Path(tempfile.gettempdir()) / f"actsave-{uuid7str()}"
-            log.critical(f"No save_dir specified, using {save_dir=}")
+            log.warning(
+                f"ActSave: Using temporary directory {save_dir} for activations."
+            )
+        else:
+            log.info(f"ActSave enabled. Saving to {save_dir}")
         self._saver = _Saver(save_dir, lambda: self._prefixes)
 
     def disable(self):
@@ -307,6 +312,18 @@ class ActSaveProvider:
         self._prefixes = []
         self._disable_count = 0
 
+        # Check for environment variable `ACTSAVE` to automatically enable saving.
+        # If set to "1" or "true" (case-insensitive), activations are saved to a temporary directory.
+        # If set to a path, activations are saved to that path.
+        if env_var := os.environ.get("ACTSAVE"):
+            log.info(
+                f"`ACTSAVE={env_var}` detected, attempting to auto-enable activation saving."
+            )
+            if env_var.lower() in ("1", "true"):
+                self.enable()
+            else:
+                self.enable(Path(env_var))
+
     @contextlib.contextmanager
     def disabled(self, condition: bool | Callable[[], bool] = True):
         """

{nshutils-0.31.0.dist-info → nshutils-0.31.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: nshutils
-Version: 0.31.0
+Version: 0.31.1
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshutils-0.31.0.dist-info → nshutils-0.31.1.dist-info}/RECORD

@@ -2,7 +2,7 @@ nshutils/__init__.py,sha256=AFx1d5k34MyJ2kCHQL5vrZB8GDp2nYUaIUEjszSa25I,477
 nshutils/__init__.pyi,sha256=R4TIk--jAgVyTibdgezJQTMce3HpMCNakAJeaDqA6bc,676
 nshutils/actsave/__init__.py,sha256=hAVsog9d1g3_rQN1TRslrl6sK1PhCGbjy8PPUAmJI58,203
 nshutils/actsave/_loader.py,sha256=btLSQdErpTmK6VyG8PxJrJNsztzyavSF71n4Ec3_49E,7619
-nshutils/actsave/_saver.py,sha256=IS9TVP8WUizoj5fHrQ6hodtjidT__LDRwz5aoWHupVo,12013
+nshutils/actsave/_saver.py,sha256=_qkX0NZYvy31hdlyfhneac4kUNS_44XjOG0ZtKpdqrg,12720
 nshutils/collections.py,sha256=QWGyANmo4Efq4XRNHDSTE9tRLStwEZHGwE0ATHR-Vqo,5233
 nshutils/display.py,sha256=Ge63yllx7gi-MKL3mKQeQ5doql_nj56-o5aoTVmusDg,1473
 nshutils/logging.py,sha256=78pv3-I_gmbKSf5_mYYBr6_H4GNBGErghAdhH9wfYIc,2205
@@ -16,6 +16,6 @@ nshutils/lovely/torch_.py,sha256=J1pDJY1zzEANqa6EaJpG1pc_SYgM8YWOo1TjWdVeiA0,294
 nshutils/lovely/utils.py,sha256=2ksT5YGVViFuWc8jSkwVCsABripJmyVJdEDDH7aab70,10459
 nshutils/snoop.py,sha256=7d7_Q5sJmINL1J29wcnxEvpV95zvZYNoVn5frCq-rww,7393
 nshutils/typecheck.py,sha256=Gi7xtfilN_UwZ1FTFqBVKDhcQzBEDonVxIv3bUj-uXY,5582
-nshutils-0.31.0.dist-info/METADATA,sha256=jfKUJjGuT8R6E09S_MVkV0-_GQilY4J50FxPE9KsowA,4406
-nshutils-0.31.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-nshutils-0.31.0.dist-info/RECORD,,
+nshutils-0.31.1.dist-info/METADATA,sha256=pW-XE-rF3TPtok9VPan8cG6IlaEjlX55NBL4SKaNOHQ,4406
+nshutils-0.31.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+nshutils-0.31.1.dist-info/RECORD,,