Merge pull request #33 from osehmathias/patch-1

update: replace @beartype any with Any in ConformerWrapper
lucidrains · May 4, 2024 · 3127e3c · 3127e3c
2 parents fbd8333 + 017d44f
commit 3127e3c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/soundstorm_pytorch/soundstorm.py b/soundstorm_pytorch/soundstorm.py
@@ -15,7 +15,7 @@
 
 from beartype import beartype
 from beartype.door import is_bearable
-from beartype.typing import Union, Dict, Optional, List, Optional
+from beartype.typing import Union, Dict, Optional, List, Optional, Any
 
 from soundstorm_pytorch.attend import Attend
 
@@ -494,7 +494,7 @@ def __init__(
         *,
         codebook_size,
         num_quantizers,
-        conformer: Union[Conformer, Dict[str, any]],
+        conformer: Union[Conformer, Dict[str, Any]],
         grouped_quantizers = 1
     ):
         super().__init__()