update stainnorm.py

TissueImageAnalytics · Feb 16, 2024 · 152b5d8 · 152b5d8
1 parent 04ad5be
commit 152b5d8
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/tiatoolbox/tools/stainnorm.py b/tiatoolbox/tools/stainnorm.py
@@ -367,13 +367,7 @@ def get_mean_std(
 def get_normalizer(
     method_name: str,
     stain_matrix: np.ndarray | None = None,
-) -> (
-    ReinhardNormalizer
-    | RuifrokNormalizer
-    | MacenkoNormalizer
-    | VahadaneNormalizer
-    | CustomNormalizer
-):
+) -> StainNormalizer:
     """Return a :class:`.StainNormalizer` with corresponding name.
 
     Args: