From ca485024ec12242839a125f829cb831c1d973fd9 Mon Sep 17 00:00:00 2001
From: Simonas <20096648+simjak@users.noreply.github.com>
Date: Wed, 13 Dec 2023 11:17:41 +0200
Subject: [PATCH] nltk download to tests
---
.github/workflows/test.yml | 3 +++
coverage.xml | 2 +-
tests/unit/test_layer.py | 4 ----
3 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 616eaea1..639071a4 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -36,6 +36,9 @@ jobs:
- name: Install dependencies
run: |
poetry install
+ - name: Download nltk data
+ run: |
+ python -m nltk.downloader punkt
- name: Pytest
run: |
make test
diff --git a/coverage.xml b/coverage.xml
index 5a9baac4..3c9c2e7c 100644
--- a/coverage.xml
+++ b/coverage.xml
@@ -1,5 +1,5 @@
<?xml version="1.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">
+<coverage version="7.3.2" timestamp="1702458856280" 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 4a527a41..4d919f91 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -1,5 +1,4 @@
import pytest
-import nltk
from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder
from semantic_router.layer import (
@@ -7,11 +6,8 @@ from semantic_router.layer import (
HybridDecisionLayer,
)
-# 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