Merge pull request #23 from Chokyotager/patch-1

[refactor] basis_keys to list
lucidrains · Nov 17, 2024 · 1a474e1 · 1a474e1
2 parents 3d5a1b5 + 4e4a27b
commit 1a474e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/equiformer_pytorch/equiformer_pytorch.py b/equiformer_pytorch/equiformer_pytorch.py
@@ -1057,7 +1057,7 @@ def basis(self):
 
     @basis.setter
     def basis(self, basis):
-        self.basis_keys = basis.keys()
+        self.basis_keys = list(basis.keys())
 
         for k, v in basis.items():
             self.register_buffer(f'basis:{k}', v)