nshutils 0.11.2__tar.gz → 0.12.2__tar.gz

{nshutils-0.11.2 → nshutils-0.12.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshutils
-Version: 0.11.2
+Version: 0.12.2
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com
@@ -12,8 +12,11 @@ Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: extra
 Requires-Dist: beartype
 Requires-Dist: jaxtyping
+Requires-Dist: lovely-numpy ; extra == "extra"
+Requires-Dist: lovely-tensors ; extra == "extra"
 Requires-Dist: numpy
 Requires-Dist: pysnooper ; extra == "extra"
+Requires-Dist: rich ; extra == "extra"
 Requires-Dist: typing-extensions
 Requires-Dist: uuid7
 Project-URL: homepage, https://github.com/nimashoghi/nshutils
@@ -35,7 +38,7 @@ pip install nshutils
 
 ### Typechecking
 
-`nshutils` provides a simple way to typecheck your code using the `jaxtyping` library. Simply call `typecheck_this_module()` at the top of your module (i.e., in the root `__init__.py` file) to enable typechecking for the entire module:
+`nshutils` provides a simple way to typecheck your code using the [`jaxtyping`](https://github.com/patrick-kidger/jaxtyping) library. Simply call `typecheck_this_module()` at the top of your module (i.e., in the root `__init__.py` file) to enable typechecking for the entire module:
 
 ```python
 from nshutils.typecheck import typecheck_this_module

{nshutils-0.11.2 → nshutils-0.12.2}/README.md

@@ -14,7 +14,7 @@ pip install nshutils
 
 ### Typechecking
 
-`nshutils` provides a simple way to typecheck your code using the `jaxtyping` library. Simply call `typecheck_this_module()` at the top of your module (i.e., in the root `__init__.py` file) to enable typechecking for the entire module:
+`nshutils` provides a simple way to typecheck your code using the [`jaxtyping`](https://github.com/patrick-kidger/jaxtyping) library. Simply call `typecheck_this_module()` at the top of your module (i.e., in the root `__init__.py` file) to enable typechecking for the entire module:
 
 ```python
 from nshutils.typecheck import typecheck_this_module

{nshutils-0.11.2 → nshutils-0.12.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshutils"
-version = "0.11.2"
+version = "0.12.2"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"
@@ -15,7 +15,10 @@ jaxtyping = "*"
 typing-extensions = "*"
 beartype = "*"
 uuid7 = "*"
+lovely-numpy = { version = "*", optional = true }
+lovely-tensors = { version = "*", optional = true }
 pysnooper = { version = "*", optional = true }
+rich = { version = "*", optional = true }
 
 [tool.poetry.group.dev.dependencies]
 pyright = "^1.1.373"
@@ -39,4 +42,4 @@ reportPrivateImportUsage = false
 ignore = ["F722", "F821", "E731", "E741"]
 
 [tool.poetry.extras]
-extra = ["pysnooper"]
+extra = ["pysnooper", "lovely-numpy", "lovely-tensors", "rich"]

{nshutils-0.11.2 → nshutils-0.12.2}/src/nshutils/snoop.py

@@ -16,8 +16,8 @@ try:
     import warnings
     from contextlib import nullcontext
 
-    import pysnooper
-    import pysnooper.utils
+    import pysnooper  # type: ignore
+    import pysnooper.utils  # type: ignore
     from pkg_resources import DistributionNotFound, get_distribution
 
     try:
@@ -165,7 +165,7 @@ try:
                     + self.tensor_format(x.QR)
                     + ")"
                 )
-            warnings.warn("Unknown return_types encountered, open a bug report!")
+            raise ValueError("Unknown return_types encountered, open a bug report!")
 
         def compute_repr(self, x):
             orig_repr_func = pysnooper.utils.get_repr_function(x, self.orig_custom_repr)