fix: minor naming fixes in transformations

Author: okunator

cellseg_models_pytorch/transforms/albu_transforms/__init__.py

@@ -5,7 +5,7 @@
     ContourTransform,
     DistTransform,
     EdgeWeightTransform,
-    HoVerNetTransform,
+    HoverNetTransform,
     OmniposeTransform,
     SmoothDistTransform,
     StardistTransform,
@@ -25,7 +25,7 @@
     "PercentileNormalization",
     "Normalization",
     "CellposeTransform",
-    "HoVerNetTransform",
+    "HoverNetTransform",
     "OmniposeTransform",
     "StardistTransform",
     "SmoothDistTransform",

cellseg_models_pytorch/transforms/albu_transforms/_composition.py

@@ -64,3 +64,7 @@ def __call__(self, **data):
             return masks
 
         return res
+
+    def __repr__(self) -> str:
+        """Return a string representation of the class."""
+        return f"{self.__class__.__name__}({self.names})"

cellseg_models_pytorch/transforms/albu_transforms/inst_transforms.py

@@ -17,7 +17,7 @@
 
 __all__ = [
     "CellposeTransform",
-    "HoVerNetTransform",
+    "HoverNetTransform",
     "OmniposeTransform",
     "StardistTransform",
     "SmoothDistTransform",
@@ -60,7 +60,7 @@ def __call__(self, inst: np.ndarray, **kwargs) -> np.ndarray:
         return gen_flow_maps(inst)
 
 
-class HoVerNetTransform(OnlyInstMapTransform):
+class HoverNetTransform(OnlyInstMapTransform):
     def __init__(self, deduplicate: bool = True) -> None:
         """Generate horizontal and vertical gradients from a label mask.