File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments