Fix tests

eigenvivek · Aug 14, 2023 · 7bde740 · 7bde740
1 parent 7c39a0d
commit 7bde740
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/notebooks/index.ipynb b/notebooks/index.ipynb
@@ -117,7 +117,7 @@
     "\n",
     "# Read in the volume and get the isocenter\n",
     "volume, spacing = load_example_ct()\n",
-    "translations = torch.tensor([[bx, by, bz]], device=device)\n",
+    "bx, by, bz = torch.tensor(volume.shape) * torch.tensor(spacing) / 2\n",
     "\n",
     "# Initialize the DRR module for generating synthetic X-rays\n",
     "device = \"cuda\" if torch.cuda.is_available() else \"cpu\"\n",
@@ -131,7 +131,7 @@
     "\n",
     "# Set the camera pose with rotations (yaw, pitch, roll) and translations (x, y, z)\n",
     "rotations = torch.tensor([[torch.pi, 0.0, torch.pi / 2]], device=device)\n",
-    "bx, by, bz = torch.tensor(volume.shape) * torch.tensor(spacing) / 2\n",
+    "translations = torch.tensor([[bx, by, bz]], device=device)\n",
     "\n",
     "# Make the DRR\n",
     "img = drr(rotations, translations)\n",

diff --git a/notebooks/tutorials/.notest b/notebooks/tutorials/.notest