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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyper-connections
-Version: 0.1.2
+Version: 0.1.5
 Summary: Hyper-Connections
 Project-URL: Homepage, https://pypi.org/project/hyper-connections/
 Project-URL: Repository, https://github.com/lucidrains/hyper-connections

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

@@ -68,7 +68,6 @@ class RMSNorm(Module):
 # residual base class
 
 class Residual(Module):
-    @beartype
     def __init__(
         self,
         *args,
@@ -119,7 +118,6 @@ class Residual(Module):
 # hyper connection residual streams
 
 class HyperConnections(Module):
-    @beartype
     def __init__(
         self,
         num_residual_streams,

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

@@ -1,5 +1,5 @@
 from __future__ import annotations
-from typing import Callable
+from typing import Callable, Union
 
 from functools import partial
 from random import randrange
@@ -142,7 +142,7 @@ class Residual(Module):
 
 # hyper connection with multiple input streams
 
-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
+InputPathType = Union[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):
     def __init__(

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

@@ -1,6 +1,6 @@
 [project]
 name = "hyper-connections"
-version = "0.1.2"
+version = "0.1.5"
 description = "Hyper-Connections"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }