assert passes on my pc

PTB-MR · Jan 20, 2025 · 0716b42 · 0716b42
1 parent 6b08e95
commit 0716b42
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/notebooks/direct_reconstruction.ipynb b/examples/notebooks/direct_reconstruction.ipynb
@@ -375,8 +375,8 @@
    "outputs": [],
    "source": [
     "# If the assert statement did not raise an exception, the results are equal.\n",
-    "assert torch.allclose(img.data, img_manual.data)\n",
-    "assert torch.allclose(img.data, img_more_manual.data)"
+    "assert torch.testing.assert_close(img.data, img_manual.data)\n",
+    "assert torch.testing.assert_close(img.data, img_more_manual.data)"
    ]
   }
  ],

diff --git a/examples/scripts/direct_reconstruction.py b/examples/scripts/direct_reconstruction.py
@@ -169,5 +169,5 @@
 
 # %%
 # If the assert statement did not raise an exception, the results are equal.
-assert torch.allclose(img.data, img_manual.data)
-assert torch.allclose(img.data, img_more_manual.data)
+assert torch.testing.assert_close(img.data, img_manual.data)
+assert torch.testing.assert_close(img.data, img_more_manual.data)