address #12

lucidrains · Sep 8, 2023 · d18cddd · d18cddd
1 parent 3179983
commit d18cddd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/equiformer_pytorch/equiformer_pytorch.py b/equiformer_pytorch/equiformer_pytorch.py
@@ -3,7 +3,7 @@
 from itertools import product
 from collections import namedtuple
 
-from beartype.typing import Optional, Union, Tuple
+from beartype.typing import Optional, Union, Tuple, Dict
 from beartype import beartype
 
 import torch
@@ -1001,7 +1001,7 @@ def device(self):
     @beartype
     def forward(
         self,
-        inputs: Union[Tensor, dict[int, Tensor]],
+        inputs: Union[Tensor, Dict[int, Tensor]],
         coors: Tensor,
         mask = None,
         adj_mat = None,

diff --git a/equiformer_pytorch/version.py b/equiformer_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.3.9'
+__version__ = '0.3.10'