diff --git a/deps/test-task-2025-07-17-v2/Makefile b/deps/test-task-2025-07-17-v2/Makefile index 4c64325..5317b47 100644 --- a/deps/test-task-2025-07-17-v2/Makefile +++ b/deps/test-task-2025-07-17-v2/Makefile @@ -35,6 +35,7 @@ venv: pyright: $(ENV_PATH)/bin/python3 -m pyright \ -p pyproject.toml \ + --threads 3 \ --pythonpath $(PYTHON_PATH) ruff_check: diff --git a/deps/test-task-2025-07-17-v2/python/online/fxreader/pr34/test_task_2025_07_17_v2/transform/worker.py b/deps/test-task-2025-07-17-v2/python/online/fxreader/pr34/test_task_2025_07_17_v2/transform/worker.py index 821e222..5e2f69e 100644 --- a/deps/test-task-2025-07-17-v2/python/online/fxreader/pr34/test_task_2025_07_17_v2/transform/worker.py +++ b/deps/test-task-2025-07-17-v2/python/online/fxreader/pr34/test_task_2025_07_17_v2/transform/worker.py @@ -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(