nshutils 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

nshutils/__init__.py

@@ -1,3 +1,6 @@
 from . import actsave as actsave
 from . import typecheck as typecheck
+from .logging import init_python_logging as init_python_logging
+from .logging import lovely as lovely
+from .logging import pretty as pretty
 from .snoop import snoop as snoop

nshutils/logging.py

@@ -0,0 +1,93 @@
+import logging
+from pathlib import Path
+
+
+def init_python_logging(
+    *,
+    lovely_tensors: bool = False,
+    lovely_numpy: bool = False,
+    rich: bool = False,
+    rich_tracebacks: bool = False,
+    log_level: int | str | None = logging.INFO,
+    log_save_dir: Path | None = None,
+):
+    if lovely_tensors:
+        try:
+            import lovely_tensors as _lovely_tensors
+
+            _lovely_tensors.monkey_patch()
+        except ImportError:
+            logging.warning(
+                "Failed to import `lovely_tensors`. Ignoring pretty PyTorch tensor formatting"
+            )
+
+    if lovely_numpy:
+        try:
+            import lovely_numpy as _lovely_numpy
+
+            _lovely_numpy.set_config(repr=_lovely_numpy.lovely)
+        except ImportError:
+            logging.warning(
+                "Failed to import `lovely_numpy`. Ignoring pretty numpy array formatting"
+            )
+
+    log_handlers: list[logging.Handler] = []
+    if log_save_dir:
+        log_file = log_save_dir / "logging.log"
+        log_file.touch(exist_ok=True)
+        log_handlers.append(logging.FileHandler(log_file))
+
+    if rich:
+        try:
+            from rich.logging import RichHandler
+
+            log_handlers.append(RichHandler(rich_tracebacks=rich_tracebacks))
+        except ImportError:
+            logging.warning(
+                "Failed to import rich. Falling back to default Python logging."
+            )
+
+    logging.basicConfig(
+        level=log_level,
+        format="%(message)s",
+        datefmt="[%X]",
+        handlers=log_handlers,
+    )
+
+
+def pretty(
+    *,
+    lovely_tensors: bool = True,
+    lovely_numpy: bool = True,
+    log_level: int | str | None = logging.INFO,
+    log_save_dir: Path | None = None,
+    rich_log_handler: bool = True,
+    rich_tracebacks: bool = True,
+):
+    init_python_logging(
+        lovely_tensors=lovely_tensors,
+        lovely_numpy=lovely_numpy,
+        rich=rich_log_handler,
+        log_level=log_level,
+        log_save_dir=log_save_dir,
+        rich_tracebacks=rich_tracebacks,
+    )
+
+
+def lovely(
+    *,
+    lovely_tensors: bool = True,
+    lovely_numpy: bool = True,
+    log_level: int | str | None = logging.INFO,
+    log_save_dir: Path | None = None,
+    rich_log_handler: bool = True,
+    rich_tracebacks: bool = True,
+):
+    pretty(
+        lovely_tensors=lovely_tensors,
+        lovely_numpy=lovely_numpy,
+        log_level=log_level,
+        log_save_dir=log_save_dir,
+        rich_log_handler=rich_log_handler,
+        rich_tracebacks=rich_tracebacks,
+    )

{nshutils-0.3.0.dist-info → nshutils-0.5.0.dist-info}/METADATA

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

{nshutils-0.3.0.dist-info → nshutils-0.5.0.dist-info}/RECORD

@@ -1,10 +1,11 @@
-nshutils/__init__.py,sha256=Uz8WR2N_h4lsbmPb3HwJHrzraxP30zGjH5YgLqLm3k8,104
+nshutils/__init__.py,sha256=1hxTdQ-e1ym76tdrSAaYoU4Qq5IUt1DRShi7khn7vs8,244
 nshutils/actsave/__init__.py,sha256=kafzChiViTeOY28GnwKD6yBJfiB7aSU8t1u5cZicdmc,280
 nshutils/actsave/_loader.py,sha256=fAhD32DrJa4onkYfcwc21YIeGEYzOSXCK_HVo9SZLgQ,4604
 nshutils/actsave/_saver.py,sha256=Uo7rNFce0KzGcS2Q03UYXoMzW8z9cRkgs_r1gLiW5Tc,9853
 nshutils/collections.py,sha256=tFzFqhqzTX_XI6IuciWJ0atZ6tj7ajGn1nzaLC4XIGQ,10155
+nshutils/logging.py,sha256=fw4NrlgNRq26YK7qHg5Zz2u0kMAfMEyutt0D_O-v7_8,2525
 nshutils/snoop.py,sha256=Rofv1Rd92E0LY40G3A-o9Hu0ZI73RR59wJD5l4Q3PDM,7022
 nshutils/typecheck.py,sha256=wrjL-H2f3J8V1lojXIbcwQBh3039bz3HBVgG9DINYK4,4819
-nshutils-0.3.0.dist-info/METADATA,sha256=y4R0kKMMAQuTKVkso9tcd4YuaBcKAvNj1toYUpvZGBc,571
-nshutils-0.3.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshutils-0.3.0.dist-info/RECORD,,
+nshutils-0.5.0.dist-info/METADATA,sha256=9c0G9DbpAwBrJNsIogzD_DgascoonI2qfIiY6RMocs8,571
+nshutils-0.5.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshutils-0.5.0.dist-info/RECORD,,