Skip to content

Commit c1cb059

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c02f153 commit c1cb059

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tiatoolbox/models/engine/prompt_segmentor.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
from tiatoolbox.models.models_abc import model_to
2929
from tiatoolbox.tools.patchextraction import PointsPatchExtractor
3030
from tiatoolbox.type_hints import Resolution, Units
31-
from tiatoolbox.wsicore.wsireader import VirtualWSIReader, WSIReader
31+
from tiatoolbox.wsicore.wsireader import VirtualWSIReader
3232

3333
if TYPE_CHECKING: # pragma: no cover
3434
from tiatoolbox.type_hints import Callable, IntBounds, IntPair
@@ -40,15 +40,15 @@ class PromptSegmentor(SemanticSegmentor):
4040
Uses the SAM2 model architecture.
4141
4242
Args:
43-
model (SAM):
43+
model (SAM):
4444
Model architecture to use.
45-
batch_size (int):
45+
batch_size (int):
4646
Batch size for processing.
47-
num_loader_workers (int):
47+
num_loader_workers (int):
4848
Number of workers for data loading.
49-
dataset_class (Callable):
49+
dataset_class (Callable):
5050
Dataset class to use.
51-
51+
5252
"""
5353

5454
def __init__(

0 commit comments

Comments
 (0)