black

Author: FynnBe

example/model_usage.ipynb

@@ -95,8 +95,13 @@
     "# Improve readiblity of validation errors\n",
     "enable_pretty_validation_errors_in_ipynb()\n",
     "\n",
+    "\n",
     "# Function to display input and prediction output images\n",
-    "def show_images(images: Union[Mapping[str, NDArray[Any]], Mapping[TensorId, Union[Tensor, NDArray[Any]]]]) -> None:\n",
+    "def show_images(\n",
+    "    images: Union[\n",
+    "        Mapping[str, NDArray[Any]], Mapping[TensorId, Union[Tensor, NDArray[Any]]]\n",
+    "    ],\n",
+    ") -> None:\n",
     "\n",
     "    for title, image in images.items():\n",
     "        if isinstance(image, Tensor):\n",
@@ -162,8 +167,8 @@
    "outputs": [],
    "source": [
     "model_source = \"affable-shark/1.2\"\n",
-    "#model_source = \"10.5281/zenodo.6287342\"\n",
-    "#model_source = \"https://uk1s3.embassy.ebi.ac.uk/public-datasets/bioimage.io/affable-shark/draft/files/rdf.yaml\""
+    "# model_source = \"10.5281/zenodo.6287342\"\n",
+    "# model_source = \"https://uk1s3.embassy.ebi.ac.uk/public-datasets/bioimage.io/affable-shark/draft/files/rdf.yaml\""
    ]
   },
   {
@@ -220,8 +225,8 @@
     "from bioimageio.core import test_model\n",
     "\n",
     "model = load_model_description(model_source, format_version=\"latest\")\n",
-    "test_summary = test_model(model) #Test model to check environment compatibility  \n",
-    "assert test_summary.status == \"passed\", test_summary.display() "
+    "test_summary = test_model(model)  # Test model to check environment compatibility\n",
+    "assert test_summary.status == \"passed\", test_summary.display()"
    ]
   },
   {
@@ -345,7 +350,6 @@
     }
    ],
    "source": [
-    "\n",
     "print(\n",
     "    f\"Model '{model.name}' requires {len(model.inputs)} input(s) with the following features:\"\n",
     ")\n",
@@ -422,7 +426,7 @@
     "prediction: Sample = predict(model=model, inputs=input)\n",
     "\n",
     "# show the input and prediction result\n",
-    "show_images(input | prediction.members)\n"
+    "show_images(input | prediction.members)"
    ]
   },
   {
@@ -484,9 +488,9 @@
     "from bioimageio.core.stat_measures import DatasetMean\n",
     "\n",
     "sample = Sample(\n",
-    "    members={\"raw\": test_input_tensor}, \n",
-    "    stat={DatasetMean(member_id=TensorId(\"raw\")): 123.456}, # Example mean value\n",
-    "    id=\"sample-from-numpy\"\n",
+    "    members={\"raw\": test_input_tensor},\n",
+    "    stat={DatasetMean(member_id=TensorId(\"raw\")): 123.456},  # Example mean value\n",
+    "    id=\"sample-from-numpy\",\n",
     ")\n",
     "\n",
     "sample"