fix(datamodule): forgot classes from lizard module

Author: okunator

cellseg_models_pytorch/datamodules/lizard_datamodule.py

@@ -122,6 +122,19 @@ def __init__(
 
         self.dataset_type = dataset_type
 
+    @property
+    def type_classes(self) -> Dict[str, int]:
+        """Pannuke cell type classes."""
+        return {
+            "bg": 0,
+            "neutrophil": 1,
+            "epithelial": 2,
+            "lymphocyte": 3,
+            "plasma": 4,
+            "eosinophil": 5,
+            "connective": 6,
+        }
+
     @staticmethod
     def download(root: str) -> None:
         """Download the lizard dataset from online."""
@@ -144,9 +157,12 @@ def prepare_data(self, rm_orig: bool = False, do_patching: bool = True) -> None:
 
         Parameters
         ----------
-        do_patching : bool, default=True
-            Flag, whether to do patching at all. Can be used if you only want to
-            download and split the data and then work it out on your own.
+            rm_orig : bool, default=False
+                After processing all the files, If True, removes the original
+                un-processed files.
+            do_patching : bool, default=True
+                Flag, whether to do patching at all. Can be used if you only want to
+                download and split the data and then work it out on your own.
         """
         folders_found = [
             d.name

cellseg_models_pytorch/datamodules/pannuke_datamodule.py

@@ -133,6 +133,12 @@ def prepare_data(self, rm_orig: bool = True) -> None:
         1. Download pannuke folds from:
             "https://warwick.ac.uk/fac/cross_fac/tia/data/pannuke/"
         2. Pre-process and split the images and masks into train, valid and test sets.
+
+        Parameters
+        ----------
+            rm_orig : bool, default=False
+                After processing all the files, If True, removes the original
+                un-processed files.
         """
         folds_found = [
             d.name