From a863f872641ee90b3f034b1f43ab0c34e1001387 Mon Sep 17 00:00:00 2001
From: Simonas <20096648+simjak@users.noreply.github.com>
Date: Wed, 13 Dec 2023 11:12:01 +0200
Subject: [PATCH] nltk download to tests
---
Makefile | 2 +-
coverage.xml | 2 +-
tests/unit/test_layer.py | 3 +++
3 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/Makefile b/Makefile
index 573998e9..3a3c42cd 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 65441b32..5a9baac4 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 8c0c9729..4a527a41 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
--
GitLab