Skip to content

Commit d577f90

Browse files
committed
overload fix
1 parent da574f8 commit d577f90

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

src/autointent/_wrappers/embedder.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -246,11 +246,6 @@ def embed(
246246
self, utterances: list[str], task_type: TaskTypeEnum | None = None, *, return_tensors: Literal[False] = False
247247
) -> npt.NDArray[np.float32]: ...
248248

249-
@overload
250-
def embed(
251-
self, utterances: list[str], task_type: TaskTypeEnum | None = None, *, return_tensors: bool = False
252-
) -> npt.NDArray[np.float32] | torch.Tensor: ...
253-
254249
def embed(
255250
self, utterances: list[str], task_type: TaskTypeEnum | None = None, return_tensors: bool = False
256251
) -> npt.NDArray[np.float32] | torch.Tensor:

0 commit comments

Comments
 (0)