Skip to content

Commit 883d8e6

Browse files
committed
fix DummyApp in unittest
1 parent 07c1482 commit 883d8e6

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

drucker_client/test/dummy_app.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33

44

55
from drucker import Drucker
6-
from drucker.utils import PredictLabel, PredictResult, EvaluateResult
6+
from drucker.utils import PredictLabel, PredictResult, EvaluateResult, EvaluateResultDetail, EvaluateDetail
7+
from typing import List, Generator
78

89

910
class DummyApp(Drucker):
@@ -18,3 +19,6 @@ def predict(self, input: PredictLabel, option: dict = None) -> PredictResult:
1819

1920
def evaluate(self, file: bytes) -> EvaluateResult:
2021
pass
22+
23+
def get_evaluate_detail(self, file_path: str, results: List[EvaluateResultDetail]) -> Generator[EvaluateDetail, None, None]:
24+
pass

0 commit comments

Comments
 (0)