|
10 | 10 | from tiatoolbox.utils import misc |
11 | 11 | from tiatoolbox.utils.exceptions import MethodNotSupported |
12 | 12 | from tiatoolbox.wsicore import wsireader |
13 | | -from tiatoolbox.wsicore.wsireader import VirtualWSIReader, WSIReader |
14 | 13 |
|
15 | 14 |
|
16 | 15 | class PatchExtractorABC(ABC): |
@@ -114,7 +113,7 @@ def __init__( |
114 | 113 | self, |
115 | 114 | input_img: Union[str, Path, np.ndarray], |
116 | 115 | patch_size: Union[int, Tuple[int, int]], |
117 | | - input_mask: Union[str, Path, np.ndarray, WSIReader] = None, |
| 116 | + input_mask: Union[str, Path, np.ndarray, wsireader.WSIReader] = None, |
118 | 117 | resolution: Union[int, float, Tuple[float, float]] = 0, |
119 | 118 | units: str = "level", |
120 | 119 | pad_mode: str = "constant", |
@@ -235,7 +234,7 @@ def _generate_location_df(self): |
235 | 234 |
|
236 | 235 | @staticmethod |
237 | 236 | def filter_coordinates_fast( |
238 | | - mask_reader: VirtualWSIReader, |
| 237 | + mask_reader: wsireader.VirtualWSIReader, |
239 | 238 | coordinates_list: np.ndarray, |
240 | 239 | coordinate_resolution: float, |
241 | 240 | coordinate_units: str, |
@@ -330,7 +329,7 @@ def filter_coordinates_fast( |
330 | 329 |
|
331 | 330 | @staticmethod |
332 | 331 | def filter_coordinates( |
333 | | - mask_reader: VirtualWSIReader, |
| 332 | + mask_reader: wsireader.VirtualWSIReader, |
334 | 333 | coordinates_list: np.ndarray, |
335 | 334 | func: Callable = None, |
336 | 335 | resolution: float = None, |
@@ -587,7 +586,7 @@ def __init__( |
587 | 586 | self, |
588 | 587 | input_img: Union[str, Path, np.ndarray], |
589 | 588 | patch_size: Union[int, Tuple[int, int]], |
590 | | - input_mask: Union[str, Path, np.ndarray, WSIReader] = None, |
| 589 | + input_mask: Union[str, Path, np.ndarray, wsireader.WSIReader] = None, |
591 | 590 | resolution: Union[int, float, Tuple[float, float]] = 0, |
592 | 591 | units: str = "level", |
593 | 592 | stride: Union[int, Tuple[int, int]] = None, |
|
0 commit comments