Skip to content

Commit 71e4ce8

Browse files
committed
try to fix unit tests
1 parent 627f8fe commit 71e4ce8

File tree

3 files changed

+19
-3
lines changed

3 files changed

+19
-3
lines changed
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
name: test embedder
2+
3+
on:
4+
push:
5+
branches:
6+
- dev
7+
pull_request:
8+
9+
jobs:
10+
test:
11+
uses: ./.github/workflows/reusable-test.yaml
12+
with:
13+
test_command: pytest -n auto tests/embedder/
14+
extras: --extra sentence-transformers
15+

.github/workflows/unit-tests.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,4 +10,4 @@ jobs:
1010
test:
1111
uses: ./.github/workflows/reusable-test.yaml
1212
with:
13-
test_command: pytest -n auto --ignore=tests/modules/scoring/ --ignore=tests/pipeline
13+
test_command: pytest -n auto --ignore=tests/modules/scoring/ --ignore=tests/pipeline --ignore=tests/embedder

tests/modules/decision/conftest.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33

44
from autointent.context.data_handler import DataHandler
55
from autointent.modules import KNNScorer
6+
from tests.conftest import get_test_embedder_config
67

78

89
@pytest.fixture
@@ -12,7 +13,7 @@ def multiclass_fit_data(dataset):
1213
knn_params = {
1314
"k": 3,
1415
"weights": "distance",
15-
"embedder_config": "sergeyzh/rubert-tiny-turbo",
16+
"embedder_config": get_test_embedder_config(),
1617
}
1718
scorer = KNNScorer(**knn_params)
1819

@@ -29,7 +30,7 @@ def multilabel_fit_data(dataset):
2930
knn_params = {
3031
"k": 3,
3132
"weights": "distance",
32-
"embedder_config": "sergeyzh/rubert-tiny-turbo",
33+
"embedder_config": get_test_embedder_config(),
3334
}
3435
scorer = KNNScorer(**knn_params)
3536

0 commit comments

Comments
 (0)