File tree Expand file tree Collapse file tree 3 files changed +19
-3
lines changed
Expand file tree Collapse file tree 3 files changed +19
-3
lines changed Original file line number Diff line number Diff line change 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+
Original file line number Diff line number Diff line change 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
Original file line number Diff line number Diff line change 33
44from autointent .context .data_handler import DataHandler
55from 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
You can’t perform that action at this time.
0 commit comments