freemt commited on
Commit
f88c21c
1 Parent(s): 72eb4f9

Update workflows to include pip install sentence-transformers

Browse files
.github/workflows/routine-tests.yml CHANGED
@@ -21,7 +21,9 @@ jobs:
21
  with:
22
  poetry-version: ${{ matrix.poetry-version }}
23
  - name: Install packages
24
- run: poetry install # --no-dev
 
 
25
  - name: Run flake8
26
  run: |
27
  poetry run flake8 --exit-zero
21
  with:
22
  poetry-version: ${{ matrix.poetry-version }}
23
  - name: Install packages
24
+ run: |
25
+ poetry install # --no-dev
26
+ poetry run pip install sentence-transformers
27
  - name: Run flake8
28
  run: |
29
  poetry run flake8 --exit-zero
radio_embed/radio_embed.py CHANGED
@@ -15,7 +15,7 @@ def radio_embed(
15
  ) -> numpy.ndarray:
16
  """Embed input."""
17
  try:
18
- _ = model.encode(text.splitlines())
19
  except Exception as _:
20
  logger.exception(_)
21
  raise
15
  ) -> numpy.ndarray:
16
  """Embed input."""
17
  try:
18
+ _ = model.encode(text.strip().splitlines())
19
  except Exception as _:
20
  logger.exception(_)
21
  raise