diff --git a/Makefile b/Makefile index 573998e954e3ae3084cb057f0d0749a58b98090d..3a3c42cd9723366e5c62544979ed22dcff209df5 100644 --- a/Makefile +++ b/Makefile @@ -11,4 +11,4 @@ lint lint_diff: poetry run ruff . test: - poetry run pytest -vv -n auto --cov=semantic_router --cov-report=term-missing --cov-report=xml --cov-fail-under=100 + poetry run pytest -vv -n 20 --cov=semantic_router --cov-report=term-missing --cov-report=xml --cov-fail-under=100 diff --git a/coverage.xml b/coverage.xml index 65441b3247366c54ae80f8dc6a4fefcc67f16016..5a9baac40ad0c645d33ae790e8d5c600a3cf7c01 100644 --- a/coverage.xml +++ b/coverage.xml @@ -1,5 +1,5 @@ <?xml version="1.0" ?> -<coverage version="7.3.2" timestamp="1702457433568" lines-valid="311" lines-covered="311" line-rate="1" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0"> +<coverage version="7.3.2" timestamp="1702458696684" lines-valid="311" lines-covered="311" line-rate="1" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0"> <!-- Generated by coverage.py: https://coverage.readthedocs.io/en/7.3.2 --> <!-- Based on https://raw.githubusercontent.com/cobertura/web/master/htdocs/xml/coverage-04.dtd --> <sources> diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 8c0c9729379e1abfa8c67278a1ef5a092c0859be..4a527a41a6f6b13dc0adebbc1279a410dc7b0104 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -1,4 +1,5 @@ import pytest +import nltk from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder from semantic_router.layer import ( @@ -9,6 +10,8 @@ from semantic_router.layer import ( # Replace with the actual module name from semantic_router.schema import Decision +nltk.download("punkt") + def mock_encoder_call(utterances): # Define a mapping of utterances to return values