[+] improve typing for worker.py
This commit is contained in:
parent
c4eb8b5568
commit
97e0270550
1
deps/test-task-2025-07-17-v2/Makefile
vendored
1
deps/test-task-2025-07-17-v2/Makefile
vendored
@ -35,6 +35,7 @@ venv:
|
||||
pyright:
|
||||
$(ENV_PATH)/bin/python3 -m pyright \
|
||||
-p pyproject.toml \
|
||||
--threads 3 \
|
||||
--pythonpath $(PYTHON_PATH)
|
||||
|
||||
ruff_check:
|
||||
|
16
deps/test-task-2025-07-17-v2/python/online/fxreader/pr34/test_task_2025_07_17_v2/transform/worker.py
vendored
16
deps/test-task-2025-07-17-v2/python/online/fxreader/pr34/test_task_2025_07_17_v2/transform/worker.py
vendored
@ -4,15 +4,15 @@ import transformers.pipelines
|
||||
from typing import (Any, cast, Callable, Protocol, Literal,)
|
||||
|
||||
class SummarizerPipeline(Protocol):
|
||||
def predict(data: str) -> str: ...
|
||||
def predict(self, data: str) -> str: ...
|
||||
|
||||
class Pipeline(Protocol):
|
||||
def __call__(
|
||||
self,
|
||||
task: Literal['summarizer'],
|
||||
task: Literal['summarization'],
|
||||
model: Any,
|
||||
tokenizer: Any,
|
||||
) -> Summarizer: ...
|
||||
) -> 'SummarizerPipeline': ...
|
||||
|
||||
class Summarizer:
|
||||
def __init__(self) -> None:
|
||||
@ -31,11 +31,11 @@ class Summarizer:
|
||||
|
||||
self.summarizer = cast(
|
||||
Pipeline,
|
||||
getattr(transformers.pipelines, 'pipeline')(
|
||||
'summarization',
|
||||
model=model,
|
||||
tokenizer=tokenizer,
|
||||
)
|
||||
getattr(transformers.pipelines, 'pipeline')
|
||||
)(
|
||||
'summarization',
|
||||
model=self.model,
|
||||
tokenizer=self.tokenizer,
|
||||
)
|
||||
|
||||
def summarize(
|
||||
|
Loading…
Reference in New Issue
Block a user