Update starter notebook (#325)

* Update starter notebook * Try increasing precision
facebookresearch · Nov 27, 2023 · 2ce690c · 2ce690c
1 parent 84ab6e5
commit 2ce690c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/fastMRI_tutorial.ipynb b/fastMRI_tutorial.ipynb
@@ -8,12 +8,10 @@
    ]
   },
   {
-   "cell_type": "code",
-   "execution_count": null,
+   "cell_type": "markdown",
    "metadata": {},
-   "outputs": [],
    "source": [
-    "# Testing if integration works"
+    "#### This notebook does not include the raw data files. In order to run the notebook, you need to download them from https://fastmri.med.nyu.edu/"
    ]
   },
   {

diff --git a/tests/test_math.py b/tests/test_math.py
@@ -25,7 +25,7 @@
 )
 def test_fft2(shape):
     shape = shape + [2]
-    x = create_input(shape)
+    x = create_input(shape).double()
     out_torch = fastmri.fft2c(x).numpy()
     out_torch = out_torch[..., 0] + 1j * out_torch[..., 1]
 
@@ -47,7 +47,7 @@ def test_fft2(shape):
 )
 def test_ifft2(shape):
     shape = shape + [2]
-    x = create_input(shape)
+    x = create_input(shape).double()
     out_torch = fastmri.ifft2c(x).numpy()
     out_torch = out_torch[..., 0] + 1j * out_torch[..., 1]