Skip to content

Commit

Permalink
tests(emebedder): use object directly
Browse files Browse the repository at this point in the history
  • Loading branch information
winstxnhdw committed Sep 23, 2024
1 parent acac5bf commit 908a3f6
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions tests/test_embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,19 +5,18 @@
from numpy import array_equal
from pytest import fixture

from server.dependencies import embedder_model
from server.features.embeddings import Embedder

type Text = Literal['Hello world!']


@fixture(scope='function')
@fixture()
def embedding() -> Iterator[Embedder]:
return embedder_model()
yield Embedder()


@fixture()
def text():
def text() -> Iterator[Text]:
yield 'Hello world!'


Expand Down

0 comments on commit 908a3f6

Please sign in to comment.