remove beartype for now

lucidrains · Jul 2, 2024 · 75ad1fd · 75ad1fd
1 parent 4bfe038
commit 75ad1fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/rotary_embedding_torch/rotary_embedding_torch.py b/rotary_embedding_torch/rotary_embedding_torch.py
@@ -8,8 +8,7 @@
 
 from einops import rearrange, repeat
 
-from beartype import beartype
-from beartype.typing import Literal
+from typing import Literal
 
 # helper functions
 
@@ -65,7 +64,6 @@ def apply_learned_rotations(rotations, t, start_index = 0, freq_ranges = None):
 # classes
 
 class RotaryEmbedding(Module):
-    @beartype
     def __init__(
         self,
         dim,
@@ -209,7 +207,6 @@ def rotate_queries_and_keys(self, q, k, seq_dim = None):
 
         return rotated_q, rotated_k
 
-    @beartype
     def get_scale(
         self,
         t: Tensor,

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'rotary-embedding-torch',
   packages = find_packages(),
-  version = '0.6.3',
+  version = '0.6.4',
   license='MIT',
   description = 'Rotary Embedding - Pytorch',
   long_description_content_type = 'text/markdown',
@@ -16,7 +16,6 @@
     'positional embedding'    
   ],
   install_requires=[
-    'beartype',
     'einops>=0.7',
     'torch>=2.0'
   ],