We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5f972d5 commit 2eac061Copy full SHA for 2eac061
1 file changed
tests/test_integration.py
@@ -220,14 +220,14 @@ def test_topic_joining():
220
221
222
def test_refitting():
223
- model = SemanticSignalSeparation(10)
+ model = SemanticSignalSeparation(10, encoder=trf)
224
model.fit(texts, embeddings=embeddings)
225
model.refit(texts, embeddings=embeddings, n_components=20)
226
assert model.components_.shape[0] == 20
227
228
229
def test_serialization():
230
231
232
with tempfile.TemporaryDirectory() as tmp_dir:
233
model.to_disk(tmp_dir)
0 commit comments