nshutils 0.11.0__tar.gz → 0.11.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshutils
-Version: 0.11.0
+Version: 0.11.2
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com
@@ -16,6 +16,7 @@ Requires-Dist: numpy
 Requires-Dist: pysnooper ; extra == "extra"
 Requires-Dist: typing-extensions
 Requires-Dist: uuid7
+Project-URL: homepage, https://github.com/nimashoghi/nshutils
 Description-Content-Type: text/markdown
 
 # nshutils
@@ -45,10 +46,10 @@ typecheck_this_module()
 You can also use the `tassert` function to assert that a value is of a certain type:
 
 ```python
-from nshutils.typecheck import tassert, Float32
+import nshutils.typecheck as tc
 
-def my_function(x: Float32):
-    tassert(Float32, x)
+def my_function(x: tc.Float[torch.Tensor, "bsz seq len"]) -> tc.Float[torch.Tensor, "bsz seq len"]:
+    tc.tassert(tc.Float[torch.Tensor, "bsz seq len"], x)
     ...
 ```
 

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

@@ -25,10 +25,10 @@ typecheck_this_module()
 You can also use the `tassert` function to assert that a value is of a certain type:
 
 ```python
-from nshutils.typecheck import tassert, Float32
+import nshutils.typecheck as tc
 
-def my_function(x: Float32):
-    tassert(Float32, x)
+def my_function(x: tc.Float[torch.Tensor, "bsz seq len"]) -> tc.Float[torch.Tensor, "bsz seq len"]:
+    tc.tassert(tc.Float[torch.Tensor, "bsz seq len"], x)
     ...
 ```
 

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

@@ -1,10 +1,13 @@
 [tool.poetry]
 name = "nshutils"
-version = "0.11.0"
+version = "0.11.2"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"
 
+[tool.poetry.urls]
+homepage = "https://github.com/nimashoghi/nshutils"
+
 [tool.poetry.dependencies]
 python = "^3.10"
 numpy = "*"