Skip to content

Commit b402c03

Browse files
committed
fix mypy
1 parent 4c8de35 commit b402c03

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

shh/services/recording.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import sys
66
from collections.abc import Callable
77

8+
import numpy as np
89
import pyperclip # type: ignore[import-untyped]
910
from numpy.typing import NDArray
1011

@@ -32,7 +33,7 @@ def __init__(self, settings: Settings) -> None:
3233
async def record_audio(
3334
self,
3435
on_progress: Callable[[float, float], None] | None = None,
35-
) -> NDArray:
36+
) -> NDArray[np.float32]:
3637
"""
3738
Record audio from the microphone until Enter is pressed or max duration reached.
3839
@@ -69,7 +70,7 @@ async def record_audio(
6970

7071
async def transcribe_and_format(
7172
self,
72-
audio_data: NDArray,
73+
audio_data: NDArray[np.float32],
7374
options: RecordingOptions,
7475
) -> TranscriptionOutput:
7576
"""

0 commit comments

Comments
 (0)