Fix syntax in reformated paddle-paddle and torch notebook snippets (#…

…5523) * Fix paddle paddle external input example * Fix torch example external input example Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com> --------- Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
NVIDIA · Jun 14, 2024 · 242d65a · 242d65a
1 parent 755439d
commit 242d65a
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 14 deletions.
diff --git a/docs/examples/frameworks/paddle/paddle-external_input.ipynb b/docs/examples/frameworks/paddle/paddle-external_input.ipynb
@@ -166,19 +166,23 @@
     }
    ],
    "source": [
-    "from nvidia.dali.plugin.paddle import DALIClassificationIterator as PaddleIterator\n",
+    "from nvidia.dali.plugin.paddle import (\n",
+    "    DALIClassificationIterator as PaddleIterator,\n",
+    ")\n",
     "from nvidia.dali.plugin.paddle import LastBatchPolicy\n",
     "\n",
     "eii = ExternalInputIterator(batch_size, 0, 1)\n",
-    "pipe = ExternalSourcePipeline(batch_size=batch_size, num_threads=2, device_id=0,\n",
-    "                              external_data=eii)\n",
-    "pii = PaddleIterator(pipe, last_batch_padded=True,\n",
-    "                     last_batch_policy=LastBatchPolicy.PARTIAL)\n",
+    "pipe = ExternalSourcePipeline(\n",
+    "    batch_size=batch_size, num_threads=2, device_id=0, external_data=eii\n",
+    ")\n",
+    "pii = PaddleIterator(\n",
+    "    pipe, last_batch_padded=True, last_batch_policy=LastBatchPolicy.PARTIAL\n",
+    ")\n",
     "\n",
     "for e in range(epochs):\n",
     "    for i, data in enumerate(pii):\n",
-    "        print(f\"epoch: {e}, iter {i}, real batch size: \"\n",
-    "              f\"{len(np.array(data[0][\"data\"]))}\")\n",
+    "        real_batch_size = len(np.array(data[0][\"data\"]))\n",
+    "        print(f\"epoch: {e}, iter {i}, real batch size: {real_batch_size}\")\n",
     "    pii.reset()"
    ]
   }

diff --git a/docs/examples/frameworks/pytorch/pytorch-external_input.ipynb b/docs/examples/frameworks/pytorch/pytorch-external_input.ipynb
@@ -169,19 +169,23 @@
     }
    ],
    "source": [
-    "from nvidia.dali.plugin.pytorch import DALIClassificationIterator as PyTorchIterator\n",
+    "from nvidia.dali.plugin.pytorch import (\n",
+    "    DALIClassificationIterator as PyTorchIterator,\n",
+    ")\n",
     "from nvidia.dali.plugin.pytorch import LastBatchPolicy\n",
     "\n",
     "eii = ExternalInputIterator(batch_size, 0, 1)\n",
-    "pipe = ExternalSourcePipeline(batch_size=batch_size, num_threads=2, device_id=0,\n",
-    "                              external_data=eii)\n",
-    "pii = PyTorchIterator(pipe, last_batch_padded=True,\n",
-    "                      last_batch_policy=LastBatchPolicy.PARTIAL)\n",
+    "pipe = ExternalSourcePipeline(\n",
+    "    batch_size=batch_size, num_threads=2, device_id=0, external_data=eii\n",
+    ")\n",
+    "pii = PyTorchIterator(\n",
+    "    pipe, last_batch_padded=True, last_batch_policy=LastBatchPolicy.PARTIAL\n",
+    ")\n",
     "\n",
     "for e in range(epochs):\n",
     "    for i, data in enumerate(pii):\n",
-    "        print(f\"epoch: {e}, iter {i}, real batch size: \"\n",
-    "              f\"{len(data[0][\"data\"])}\")\n",
+    "        real_batch_size = len(data[0][\"data\"])\n",
+    "        print(f\"epoch: {e}, iter {i}, real batch size: {real_batch_size}\")\n",
     "    pii.reset()"
    ]
   }