hyper-connections 0.1.0__tar.gz → 0.1.2__tar.gz

{hyper_connections-0.1.0 → hyper_connections-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyper-connections
-Version: 0.1.0
+Version: 0.1.2
 Summary: Hyper-Connections
 Project-URL: Homepage, https://pypi.org/project/hyper-connections/
 Project-URL: Repository, https://github.com/lucidrains/hyper-connections
@@ -34,12 +34,9 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9
-Requires-Dist: beartype
 Requires-Dist: einops>=0.8.0
 Requires-Dist: torch>=2.3
 Provides-Extra: examples
-Provides-Extra: test
-Requires-Dist: pytest; extra == 'test'
 Description-Content-Type: text/markdown
 
 <img src="./hyper-connections.png" width="450px"></img>

{hyper_connections-0.1.0 → hyper_connections-0.1.2}/hyper_connections/hyper_connections.py

@@ -12,8 +12,6 @@ from torch.utils._pytree import tree_flatten, tree_unflatten
 
 from einops import rearrange, repeat, reduce, einsum
 
-from beartype import beartype
-
 """
 ein notation:
 b - batch

{hyper_connections-0.1.0 → hyper_connections-0.1.2}/hyper_connections/hyper_connections_with_multi_branch_inputs.py

@@ -12,8 +12,6 @@ from torch.utils._pytree import tree_flatten, tree_unflatten
 
 from einops import rearrange, repeat, reduce, einsum
 
-from beartype import beartype
-
 """
 ein notation:
 b - batch
@@ -65,7 +63,6 @@ def get_init_and_expand_reduce_stream_functions(cls, num_streams, disable = Fals
 # hyper connection residual streams
 
 class HyperConnections(Module):
-    @beartype
     def __init__(
         self,
         num_residual_streams,

{hyper_connections-0.1.0 → hyper_connections-0.1.2}/hyper_connections/hyper_connections_with_multi_input_streams.py

@@ -13,8 +13,6 @@ from torch.utils._pytree import tree_flatten, tree_unflatten
 from einops import rearrange, repeat, reduce, einsum
 from einops.layers.torch import Rearrange
 
-from beartype import beartype
-
 """
 ein notation:
 b - batch
@@ -95,7 +93,6 @@ class ProjActScale(Module):
 # residual base class
 
 class Residual(Module):
-    @beartype
     def __init__(
         self,
         *args,
@@ -148,7 +145,6 @@ class Residual(Module):
 InputPathType = int | str  # the path to the second residual stream, where `int` points to *args[`int`] and `str` points to **kwargs[`str`] - `int` needs to be > 0, as 0 is the default input residual stream
 
 class HyperConnections(Module):
-    @beartype
     def __init__(
         self,
         num_residual_streams,

{hyper_connections-0.1.0 → hyper_connections-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hyper-connections"
-version = "0.1.0"
+version = "0.1.2"
 description = "Hyper-Connections"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }
@@ -23,7 +23,6 @@ classifiers=[
 ]
 
 dependencies = [
-    "beartype",
     "einops>=0.8.0",
     "torch>=2.3",
 ]
@@ -34,14 +33,9 @@ Repository = "https://github.com/lucidrains/hyper-connections"
 
 [project.optional-dependencies]
 examples = []
-test = [
-    "pytest"
-]
 
 [tool.pytest.ini_options]
-pythonpath = [
-  "."
-]
+pythonpath = ["."]
 
 [build-system]
 requires = ["hatchling"]
@@ -49,7 +43,7 @@ build-backend = "hatchling.build"
 
 [tool.rye]
 managed = true
-dev-dependencies = []
+dev-dependencies = ["pytest>=8.2.0"]
 
 [tool.hatch.metadata]
 allow-direct-references = true