Skip to content

Commit 34e0358

Browse files
Removed dedicated cross-lingual test and added it to general integration test
1 parent 3f349db commit 34e0358

1 file changed

Lines changed: 1 addition & 12 deletions

File tree

tests/test_integration.py

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ def generate_dates(
6262
GMM(3, encoder=trf),
6363
SemanticSignalSeparation(3, encoder=trf),
6464
KeyNMF(3, encoder=trf),
65+
KeyNMF(3, encoder=trf, cross_lingual=True),
6566
ClusteringTopicModel(
6667
dimensionality_reduction=PCA(10),
6768
clustering=KMeans(3),
@@ -225,15 +226,3 @@ def test_serialization():
225226
with tempfile.TemporaryDirectory() as tmp_dir:
226227
model.to_disk(tmp_dir)
227228
model = load_model(tmp_dir)
228-
229-
230-
def test_cross_lingual():
231-
from datasets import load_dataset
232-
233-
ds = load_dataset(
234-
"aiana94/polynews-parallel", "dan_Latn-hun_Latn", split="train"
235-
)
236-
corpus = list(ds["src"]) + list(ds["tgt"])
237-
model = KeyNMF(5, cross_lingual=True)
238-
model.fit(corpus)
239-
model.print_topics()

0 commit comments

Comments
 (0)