Skip to content

Commit 4473605

Browse files
committed
chore: fix ruff findings
1 parent 6c4442c commit 4473605

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

bertopic/plotting/_approximate_distribution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def visualize_approximate_distribution(
7575
df = pd.DataFrame(topic_token_distribution).T
7676

7777
df.columns = [f"{token}_{i}" for i, token in enumerate(tokens)]
78-
df.columns = [f"{token}{' '*i}" for i, token in enumerate(tokens)]
78+
df.columns = [f"{token}{' ' * i}" for i, token in enumerate(tokens)]
7979
df.index = list(topic_model.topic_labels_.values())[topic_model._outliers :]
8080
df = df.loc[(df.sum(axis=1) != 0), :]
8181

tests/test_other.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ def test_load_save_model():
55
model = BERTopic(language="Dutch", embedding_model=None)
66
model.save("test", serialization="pickle")
77
loaded_model = BERTopic.load("test")
8-
assert type(model) == type(loaded_model)
8+
assert type(model) is type(loaded_model)
99
assert model.language == loaded_model.language
1010
assert model.embedding_model == loaded_model.embedding_model
1111
assert model.top_n_words == loaded_model.top_n_words

0 commit comments

Comments
 (0)