[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

acceleration/fast_inference_tutorial/fast_inference_tutorial.ipynb

@@ -370,7 +370,7 @@
     "    total_time_dict = {}\n",
     "    roi_size = (96, 96, 96)\n",
     "    sw_batch_size = 4\n",
-    "    \n",
+    "\n",
     "    for idx, sample in enumerate(data_list[:10]):\n",
     "        start = timer()\n",
     "        data = infer_transforms({\"image\": sample})\n",
@@ -474,7 +474,9 @@
     "    all_df = pd.merge(all_df, df, on=\"file_name\", how=\"left\")\n",
     "\n",
     "# for each file, add it's size\n",
-    "all_df[\"file_size\"] = all_df[\"file_name\"].apply(lambda x: os.path.getsize(os.path.join(root_dir, \"Task03_Liver\", \"imagesTs_nii\", x)))\n",
+    "all_df[\"file_size\"] = all_df[\"file_name\"].apply(\n",
+    "    lambda x: os.path.getsize(os.path.join(root_dir, \"Task03_Liver\", \"imagesTs_nii\", x))\n",
+    ")\n",
     "# sort by file size\n",
     "all_df = all_df.sort_values(by=\"file_size\", ascending=True)\n",
     "# convert file size to MB\n",
@@ -538,8 +540,14 @@
    ],
    "source": [
     "print(\"TensorRT Improvement: \", (total_time[\"original_time\"] - total_time[\"trt_time\"]) / total_time[\"original_time\"])\n",
-    "print(\"TensorRT + GPU Transforms Improvement: \", (total_time[\"original_time\"] - total_time[\"trt_gpu_transforms_time\"]) / total_time[\"original_time\"])\n",
-    "print(\"TensorRT + GDS + GPU Transforms Improvement: \", (total_time[\"original_time\"] - total_time[\"trt_gds_gpu_transforms_time\"]) / total_time[\"original_time\"])"
+    "print(\n",
+    "    \"TensorRT + GPU Transforms Improvement: \",\n",
+    "    (total_time[\"original_time\"] - total_time[\"trt_gpu_transforms_time\"]) / total_time[\"original_time\"],\n",
+    ")\n",
+    "print(\n",
+    "    \"TensorRT + GDS + GPU Transforms Improvement: \",\n",
+    "    (total_time[\"original_time\"] - total_time[\"trt_gds_gpu_transforms_time\"]) / total_time[\"original_time\"],\n",
+    ")"
    ]
   },
   {
@@ -562,10 +570,10 @@
     "total_time.index = [\"pytorch_model\", \"TensorRT\", \"TensorRT_GPU_Transform\", \"TensorRT_GPU_Transform_GDS\"]\n",
     "\n",
     "plt.figure(figsize=(10, 6))\n",
-    "total_time.plot(kind='bar', color=['skyblue', 'orange', 'green', 'red'])\n",
-    "plt.title('Total Inference Time for Each Benchmark Type')\n",
-    "plt.xlabel('Benchmark Type')\n",
-    "plt.ylabel('Total Time (seconds)')\n",
+    "total_time.plot(kind=\"bar\", color=[\"skyblue\", \"orange\", \"green\", \"red\"])\n",
+    "plt.title(\"Total Inference Time for Each Benchmark Type\")\n",
+    "plt.xlabel(\"Benchmark Type\")\n",
+    "plt.ylabel(\"Total Time (seconds)\")\n",
     "plt.xticks(rotation=45)\n",
     "plt.tight_layout()\n",
     "plt.show()"