nanom commited on
Commit
59c4770
1 Parent(s): 3bafa16

Fix new error in barplot inside plot_projection_scores function

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