nanom commited on
Commit
b9cbef6
1 Parent(s): 639eed4

Fix new error in barplot inside plot_projection_scores function

Browse files
Files changed (1) hide show
  1. modules/module_BiasExplorer.py +5 -2
modules/module_BiasExplorer.py CHANGED
@@ -362,8 +362,11 @@ class WEBiasExplorer2Spaces(WordBiasExplorer):
362
  .max(),
363
  decimals=1)
364
 
365
- sns.barplot(x='projection', y='word', data=projections_df,
366
- palette=projections_df['color'])
 
 
 
367
 
368
  plt.xticks(np.arange(-most_extream_projection,
369
  most_extream_projection + axis_projection_step,
 
362
  .max(),
363
  decimals=1)
364
 
365
+ sns.barplot(x='projection',
366
+ y='word',
367
+ data=projections_df,
368
+ palette=projections_df['color'].tolist()
369
+ )
370
 
371
  plt.xticks(np.arange(-most_extream_projection,
372
  most_extream_projection + axis_projection_step,