Add missing typing for OfflineRecognizer (#482)

* Add missing typing for OfflineRecognizer * Add missing typing for OfflineStream --------- Co-authored-by: trunglebka <[email protected]>
k2-fsa · Oct 9, 2023 · e7238cd · e7238cd
1 parent 5b44be9
commit e7238cd
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/sherpa/python/sherpa/__init__.pyi b/sherpa/python/sherpa/__init__.pyi
@@ -154,14 +154,24 @@ class OfflineRecognitionResult:
 
 class OfflineStream:
     def accept_wave_file(self, filename: str) -> None: ...
+    @overload
+    def accept_samples(self, samples: List[float]) -> None: ...
+    @overload
     def accept_samples(self, samples: torch.Tensor) -> None: ...
-
+    @property
+    def result(self) -> OfflineRecognitionResult: ...
     accept_waveform = accept_samples
 
 class OfflineRecognizer:
     def __init__(self, config: OfflineRecognizerConfig) -> None: ...
+    @overload
     def create_stream(self) -> OfflineStream: ...
+    @overload
+    def create_stream(
+        self, contexts_list: List[List[int]]
+    ) -> OfflineStream: ...
     def decode_stream(self, s: OfflineStream) -> None: ...
+    def decode_streams(self, ss: List[OfflineStream]) -> None: ...
 
 class OnlineRecognizerConfig:
     @overload