Skip to content

Commit 2eac061

Browse files
changed encoder parameter in test
1 parent 5f972d5 commit 2eac061

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_integration.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -220,14 +220,14 @@ def test_topic_joining():
220220

221221

222222
def test_refitting():
223-
model = SemanticSignalSeparation(10)
223+
model = SemanticSignalSeparation(10, encoder=trf)
224224
model.fit(texts, embeddings=embeddings)
225225
model.refit(texts, embeddings=embeddings, n_components=20)
226226
assert model.components_.shape[0] == 20
227227

228228

229229
def test_serialization():
230-
model = SemanticSignalSeparation(10)
230+
model = SemanticSignalSeparation(10, encoder=trf)
231231
model.fit(texts, embeddings=embeddings)
232232
with tempfile.TemporaryDirectory() as tmp_dir:
233233
model.to_disk(tmp_dir)

0 commit comments

Comments
 (0)