Updated docstrings

Author: stergioc

sopa/io/reader/wsi.py

@@ -13,15 +13,15 @@
 def wsi(
     path: str | Path,
     as_image: bool = False,
-    backend: Literal["tiffslide", "openslide"] = "tiffslide",
+    backend: Literal["tiffslide", "openslide", "slideio"] = "tiffslide",
 ) -> SpatialData | DataTree:
     """Read a WSI into a `SpatialData` object
 
     Args:
         path: Path to the WSI
         chunks: Tuple representing the chunksize for the dimensions `(C, Y, X)`.
         as_image: If `True`, returns a, image instead of a `SpatialData` object
-        backend: The library to use as a backend in order to load the WSI. One of: `"openslide"`, `"tiffslide"`.
+        backend: The library to use as a backend in order to load the WSI. One of: `"openslide"`, `"tiffslide"`, `"slideio"`.
 
     Returns:
         A `SpatialData` object with a multiscale 2D-image of shape `(C, Y, X)`, or just the DataTree if `as_image=True`
@@ -70,7 +70,7 @@ def _get_scale_transformation(scale_factor: float):
 def wsi_autoscale(
     path: str | Path,
     image_model_kwargs: dict | None = None,
-    backend: Literal["tiffslide", "openslide"] = "tiffslide",
+    backend: Literal["tiffslide", "openslide", "slideio"] = "tiffslide",
 ) -> SpatialData:
     """Read a WSI into a `SpatialData` object.
 
@@ -80,7 +80,7 @@ def wsi_autoscale(
     Args:
         path: Path to the WSI
         image_model_kwargs: Kwargs provided to the `Image2DModel`
-        backend: The library to use as a backend in order to load the WSI. One of: `"openslide"`, `"tiffslide"`.
+        backend: The library to use as a backend in order to load the WSI. One of: `"openslide"`, `"tiffslide"`, `"slideio"`.
 
     Returns:
         A `SpatialData` object with a 2D-image of shape `(C, Y, X)`

sopa/patches/infer.py

@@ -50,6 +50,7 @@ def compute_embeddings(
         image_key: Optional image key of the image, unecessary if there is only one image.
         batch_size: Mini-batch size used during inference.
         device: Device used for the computer vision model.
+        data_parallel: If `True`, the model will be run in data parallel mode. If a list of GPUs is provided, the model will be run in data parallel mode on the specified GPUs.
         key_added: Optional name of the spatial element that will be added (storing the embeddings).
     """
     try: