From 152b5d8f3e27d26d25ed7d8043525e2ed7f3e865 Mon Sep 17 00:00:00 2001
From: Jiaqi Lv <lvjiaqi9@gmail.com>
Date: Fri, 16 Feb 2024 14:17:24 +0000
Subject: [PATCH] update stainnorm.py

---
 tiatoolbox/tools/stainnorm.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/tiatoolbox/tools/stainnorm.py b/tiatoolbox/tools/stainnorm.py
index 755c28d22..bad835236 100644
--- 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: