Skip to content

Commit

Permalink
Allow transform() to be None in tests
Browse files Browse the repository at this point in the history
  • Loading branch information
x-tabdeveloping committed Jul 1, 2024
1 parent c11c86d commit 9976bcc
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tests/test_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,8 @@ def test_fit_online(model):
def test_prepare_topic_data(model):
topic_data = model.prepare_topic_data(texts, embeddings=embeddings)
for key, value in topic_data.items():
# We allow transform() to be None for transductive models
if key == "transform":
continue
if value is None:
raise TypeError(
"None of the fields of prepare_topic_data should be None."
)
raise TypeError(f"Field {key} is None in topic_data.")

0 comments on commit 9976bcc

Please sign in to comment.