fix an issue with inplace detach, release 0.2.2

lucidrains · Nov 25, 2022 · 35198fb · 35198fb
1 parent 4bb3669
commit 35198fb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -27,8 +27,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
+        python -m pip install torch==1.10.0
         python -m pip install pytest
-        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Test with pytest
       run: |
         python setup.py pytest
diff --git a/invariant_point_attention/invariant_point_attention.py b/invariant_point_attention/invariant_point_attention.py
@@ -333,7 +333,7 @@ def forward(
             rotations = quaternion_to_matrix(quaternions)
 
             if self.detach_rotations:
-                rotations.detach_()
+                rotations = rotations.detach()
 
             x = block(
                 x,

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'invariant-point-attention',
   packages = find_packages(),
-  version = '0.2.1',
+  version = '0.2.2',
   license='MIT',
   description = 'Invariant Point Attention',
   long_description_content_type = 'text/markdown',