From 2ce690c583bd65e8b4bbf255025b7403349f2c52 Mon Sep 17 00:00:00 2001 From: Matthew Muckley Date: Mon, 27 Nov 2023 12:59:02 -0500 Subject: [PATCH] Update starter notebook (#325) * Update starter notebook * Try increasing precision --- fastMRI_tutorial.ipynb | 6 ++---- tests/test_math.py | 4 ++-- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/fastMRI_tutorial.ipynb b/fastMRI_tutorial.ipynb index 91bb865a..9f4489eb 100644 --- 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 index cb8debfd..c9e17ccc 100644 --- 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]