Skip to content

Commit cb32bdb

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 1a245ad commit cb32bdb

File tree

1 file changed

+18
-17
lines changed

1 file changed

+18
-17
lines changed

hugging_face/hugging_face_pipeline_for_monai.ipynb

Lines changed: 18 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -346,17 +346,21 @@
346346
" ),\n",
347347
" AsDiscreted(keys=pred_key, argmax=True),\n",
348348
" ]\n",
349-
" \n",
350-
" transform_list.append(\n",
351-
" SaveImaged(\n",
352-
" keys=pred_key,\n",
353-
" output_dir=output_dir,\n",
354-
" output_ext=output_ext,\n",
355-
" output_dtype=output_dtype,\n",
356-
" output_postfix=output_postfix,\n",
357-
" separate_folder=separate_folder,\n",
349+
"\n",
350+
" (\n",
351+
" transform_list.append(\n",
352+
" SaveImaged(\n",
353+
" keys=pred_key,\n",
354+
" output_dir=output_dir,\n",
355+
" output_ext=output_ext,\n",
356+
" output_dtype=output_dtype,\n",
357+
" output_postfix=output_postfix,\n",
358+
" separate_folder=separate_folder,\n",
359+
" )\n",
358360
" )\n",
359-
" )if save_output else transform_list\n",
361+
" if save_output\n",
362+
" else transform_list\n",
363+
" )\n",
360364
" postprocessing_transforms = Compose(transform_list)\n",
361365
" return postprocessing_transforms\n",
362366
"\n",
@@ -421,14 +425,11 @@
421425
" for key, value in kwargs.items():\n",
422426
" if key not in self.POSTPROCESSING_EXTRA_ARGS:\n",
423427
" logging.warning(f\"Cannot set parameter {key} for postprocessing.\")\n",
424-
" if (\n",
425-
" key in self._postprocess_params\n",
426-
" and value != self._postprocess_params[key]\n",
427-
" ) or (key not in self._postprocess_params):\n",
428+
" if (key in self._postprocess_params and value != self._postprocess_params[key]) or (\n",
429+
" key not in self._postprocess_params\n",
430+
" ):\n",
428431
" self._postprocess_params.update(kwargs)\n",
429-
" self.postprocessing_transforms = self._init_postprocessing_transforms(\n",
430-
" **self._postprocess_params\n",
431-
" )\n",
432+
" self.postprocessing_transforms = self._init_postprocessing_transforms(**self._postprocess_params)\n",
432433
" outputs = decollate_batch(outputs)\n",
433434
" outputs = self.postprocessing_transforms(outputs)\n",
434435
" return outputs"

0 commit comments

Comments
 (0)