nshutils 0.16.0__py3-none-any.whl → 0.18.0__py3-none-any.whl

nshutils/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from . import actsave as actsave
 from . import typecheck as typecheck
 from ._display import display as display

nshutils/_display.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import importlib.util
 from functools import cache
 from typing import Any

nshutils/actsave/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from ._loader import ActLoad as ActLoad
 from ._saver import Activation as Activation
 from ._saver import ActSave as ActSave

nshutils/actsave/_loader.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pprint
 from dataclasses import dataclass, field
 from functools import cached_property

nshutils/actsave/_saver.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import contextlib
 import fnmatch
 import tempfile

nshutils/collections.py

@@ -2,6 +2,8 @@
 # Licensed under the Apache License, Version 2.0 (the "License");
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
+from __future__ import annotations
+
 import dataclasses
 from collections import OrderedDict, defaultdict
 from collections.abc import Callable, Mapping, Sequence

nshutils/logging.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from pathlib import Path
 
@@ -7,7 +9,7 @@ def init_python_logging(
     lovely_tensors: bool = False,
     lovely_numpy: bool = False,
     treescope: bool = False,
-    treescope_autovisualize_arrays: bool = True,
+    treescope_autovisualize_arrays: bool = False,
     rich: bool = False,
     rich_tracebacks: bool = False,
     log_level: int | str | None = logging.INFO,
@@ -19,7 +21,7 @@ def init_python_logging(
 
             _lovely_tensors.monkey_patch()
         except ImportError:
-            logging.exception(
+            logging.info(
                 "Failed to import `lovely_tensors`. Ignoring pretty PyTorch tensor formatting"
             )
 
@@ -29,7 +31,7 @@ def init_python_logging(
 
             _lovely_numpy.set_config(repr=_lovely_numpy.lovely)
         except ImportError:
-            logging.exception(
+            logging.info(
                 "Failed to import `lovely_numpy`. Ignoring pretty numpy array formatting"
             )
 
@@ -45,11 +47,11 @@ def init_python_logging(
                     autovisualize_arrays=treescope_autovisualize_arrays
                 )
             else:
-                logging.exception(
+                logging.info(
                     "Treescope setup is only supported in Jupyter notebooks. Skipping."
                 )
         except ImportError:
-            logging.exception(
+            logging.info(
                 "Failed to import `treescope` or `IPython`. Ignoring `treescope` registration"
             )
 
@@ -65,7 +67,7 @@ def init_python_logging(
 
             log_handlers.append(RichHandler(rich_tracebacks=rich_tracebacks))
         except ImportError:
-            logging.exception(
+            logging.info(
                 "Failed to import rich. Falling back to default Python logging."
             )
 
@@ -81,8 +83,8 @@ def pretty(
     *,
     lovely_tensors: bool = True,
     lovely_numpy: bool = True,
-    treescope: bool = True,
-    treescope_autovisualize_arrays: bool = True,
+    treescope: bool = False,
+    treescope_autovisualize_arrays: bool = False,
     log_level: int | str | None = logging.INFO,
     log_save_dir: Path | None = None,
     rich_log_handler: bool = False,
@@ -104,8 +106,8 @@ def lovely(
     *,
     lovely_tensors: bool = True,
     lovely_numpy: bool = True,
-    treescope: bool = True,
-    treescope_autovisualize_arrays: bool = True,
+    treescope: bool = False,
+    treescope_autovisualize_arrays: bool = False,
     log_level: int | str | None = logging.INFO,
     log_save_dir: Path | None = None,
     rich_log_handler: bool = False,

nshutils/snoop.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import contextlib
 import importlib.metadata
 from typing import Any, Protocol, cast

nshutils/typecheck.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 from collections.abc import Sequence
 from logging import getLogger

{nshutils-0.16.0.dist-info → nshutils-0.18.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshutils
-Version: 0.16.0
+Version: 0.18.0
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

nshutils-0.18.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+nshutils/__init__.py,sha256=1kiWfKqSELCpTwUv4B6m43cnp2W4RPoD2DX9svgd3IA,575
+nshutils/_display.py,sha256=lfxjB5A2SMdtrtElYCP_xhdgIsPjA8XL21G5jKv6dys,1477
+nshutils/actsave/__init__.py,sha256=hAVsog9d1g3_rQN1TRslrl6sK1PhCGbjy8PPUAmJI58,203
+nshutils/actsave/_loader.py,sha256=mof3HezUNvLliz7macstX6ewXW05L0Mtv3zJyrbmImg,4640
+nshutils/actsave/_saver.py,sha256=Kor7PEk__noRDEAfCZ_I4vxql5UfWSIQIQ1OmW2RRTI,10290
+nshutils/collections.py,sha256=QWGyANmo4Efq4XRNHDSTE9tRLStwEZHGwE0ATHR-Vqo,5233
+nshutils/logging.py,sha256=-6IB0GTDDS8ue1H2tzkv_OLf4bZVN1ywL08TlDZWbtQ,3737
+nshutils/snoop.py,sha256=VuEMwFXxI1mXrv2IDaj947s3UhgvoGhgaRxyu2iiIgI,7405
+nshutils/typecheck.py,sha256=IC0cbPLlYV2Fnwv4xUl458fhye9_4O3zOpc1dNEgmVM,4951
+nshutils-0.18.0.dist-info/METADATA,sha256=TiMkBQTVYLIHbr81cg7GP-hSaTik_shtem0K8T06h1k,4117
+nshutils-0.18.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshutils-0.18.0.dist-info/RECORD,,

nshutils-0.16.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-nshutils/__init__.py,sha256=T1MnsjctEI1c5wM06d_hUOuGIYLIz_NEb1e_tT-TgQ0,539
-nshutils/_display.py,sha256=6CSaKdsFBHBGQXMppumRfO5fOfiJDXQ7FhtAvN8id7M,1441
-nshutils/actsave/__init__.py,sha256=6gJ49011Ad3kS8BejeZRPj7ZyVyIcmX-VKLEmYZyGM8,167
-nshutils/actsave/_loader.py,sha256=fAhD32DrJa4onkYfcwc21YIeGEYzOSXCK_HVo9SZLgQ,4604
-nshutils/actsave/_saver.py,sha256=-uKMmMKjEErCa8pfLhAW4077GFcUjxr0Qq0PFAnbEOw,10254
-nshutils/collections.py,sha256=EE_qLd-LrsX5lsyk9GSKh03Q8bhn9CHB3jiEeNj4uF4,5197
-nshutils/logging.py,sha256=3NwW2pSoXBktiRWPGtUzs8BrOW8H9ZpfI-8wiLbEnEE,3721
-nshutils/snoop.py,sha256=JW2JZg6oZd9WWubOP6ok6GUbh5mIQUOdP_T29iH5RCE,7369
-nshutils/typecheck.py,sha256=_KtfinRy9A0Dgq78kN5MGGrefvb6jn2tGY6svdLEBAs,4915
-nshutils-0.16.0.dist-info/METADATA,sha256=M9Mt17Jj5kzN2JMPyjLA7fLBCFgGtGivhzQ9JBS0IcQ,4117
-nshutils-0.16.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshutils-0.16.0.dist-info/RECORD,,