nshutils 0.17.0__tar.gz → 0.18.0__tar.gz

{nshutils-0.17.0 → nshutils-0.18.0}/PKG-INFO

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

{nshutils-0.17.0 → nshutils-0.18.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshutils"
-version = "0.17.0"
+version = "0.18.0"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"
@@ -40,7 +40,11 @@ strictSetInference = true
 reportPrivateImportUsage = false
 
 [tool.ruff.lint]
+select = ["FA102", "FA100"]
 ignore = ["F722", "F821", "E731", "E741"]
 
+[tool.ruff.lint.isort]
+required-imports = ["from __future__ import annotations"]
+
 [tool.poetry.extras]
 extra = ["pysnooper", "lovely-numpy", "lovely-tensors", "rich", "treescope"]

{nshutils-0.17.0 → nshutils-0.18.0}/src/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-0.17.0 → nshutils-0.18.0}/src/nshutils/_display.py

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

{nshutils-0.17.0 → nshutils-0.18.0}/src/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-0.17.0 → nshutils-0.18.0}/src/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-0.17.0 → nshutils-0.18.0}/src/nshutils/actsave/_saver.py

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

{nshutils-0.17.0 → nshutils-0.18.0}/src/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-0.17.0 → nshutils-0.18.0}/src/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,
@@ -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-0.17.0 → nshutils-0.18.0}/src/nshutils/snoop.py

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

{nshutils-0.17.0 → nshutils-0.18.0}/src/nshutils/typecheck.py

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