change doc format for patch dataset

mu_map/dataset/patches.py

@@ -11,15 +11,24 @@ class MuMapPatchDataset(MuMapDataset):
     """
     A wrapper around the MuMapDataset that computes patches for each reconstruction-μ-map pair.
 
-    :param dataset_dir: the directory containing the dataset - is passed to MuMapDataset
-    :param patches_per_image: the amount of patches to randomly generate for each image
-    :param patch_size: the size of patches in x- and y-direction
-    :param patch_size_z: the size of patches in z-direction --- it is a separate parameter because
-                         images are typically shorter in this direction
-    :param patch_offset: offset of generated patches to the border of images --- this space will
-                         then not appear in patches because it is often empty
-    :param shuffle: shuffle the patches so that patches of image pairs are mixed
-    :param **kwargs: remaining parameters passed to MuMapDataset
+    Parameters
+    ----------
+    dataset_dir: str
+        the directory containing the dataset - is passed to MuMapDataset
+    patches_per_image: int
+        the amount of patches to randomly generate for each image
+    patch_size: int
+        the size of patches in x- and y-direction
+    patch_size_z: int
+        the size of patches in z-direction --- it is a separate parameter because
+        images are typically shorter in this direction
+    patch_offset: int
+        offset of generated patches to the border of images --- this space will
+        then not appear in patches because it is often empty
+    shuffle: bool
+        shuffle the patches so that patches of image pairs are mixed
+    **kwargs:
+        remaining parameters passed to MuMapDataset
     """
 
     def __init__(
@@ -138,7 +147,7 @@ if __name__ == "__main__":
     parser.add_argument(
         "--dataset_dir",
         type=str,
-        default="data/initial/",
+        default="data/second/",
         help="the directory of the dataset",
     )
     for key, _default, _help in zip(param_keys, param_defaults, param_help):