fix transforms in notebooks

VectorInstitute · Nov 21, 2023 · 4e2de73 · 4e2de73
1 parent 7b179bc
commit 4e2de73
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/docs/source/tutorials/nihcxr/cxr_classification.ipynb b/docs/source/tutorials/nihcxr/cxr_classification.ipynb
@@ -144,8 +144,9 @@
     "            allow_missing_keys=True,\n",
     "        ),\n",
     "        Lambdad(\n",
-    "            (\"image\",),\n",
-    "            func=lambda x: np.mean(x, axis=0)[np.newaxis, :] if x.shape[0] != 1 else x,\n",
+    "            keys=(\"image\",),\n",
+    "            func=lambda x: x[0][np.newaxis, :] if x.shape[0] != 1 else x,\n",
+    "            allow_missing_keys=True,\n",
     "        ),\n",
     "    ],\n",
     ")"

diff --git a/docs/source/tutorials/nihcxr/generate_nihcxr_report.py b/docs/source/tutorials/nihcxr/generate_nihcxr_report.py
@@ -57,7 +57,7 @@
         ),
         Lambdad(
             keys=("image",),
-            func=lambda x: np.mean(x, axis=0)[np.newaxis, :] if x.shape[0] != 1 else x,
+            func=lambda x: x[0][np.newaxis, :] if x.shape[0] != 1 else x,
         ),
     ],
 )