chrisjay ThomasSimonini HF staff commited on
Commit
1ff0f9e
1 Parent(s): 34a936a

Adding str accuracy (#2)

Browse files

- Adding str accuracy (57059b967864ac80b1e91deea81348b02ea3766f)


Co-authored-by: Thomas Simonini <ThomasSimonini@users.noreply.huggingface.co>

Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -48,6 +48,7 @@ def parse_rewards(accuracy):
48
  default_std = -1000
49
  default_reward=-1000
50
  if accuracy != None:
 
51
  parsed = accuracy.split(' +/- ')
52
  if len(parsed)>1:
53
  mean_reward = float(parsed[0])
 
48
  default_std = -1000
49
  default_reward=-1000
50
  if accuracy != None:
51
+ accuracy = str(accuracy)
52
  parsed = accuracy.split(' +/- ')
53
  if len(parsed)>1:
54
  mean_reward = float(parsed[0])