bkb2135 commited on
Commit
1adf887
·
1 Parent(s): be32e3b

Use macrocrux and fill turn

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -22,7 +22,7 @@ KEYS = ['_step','_timestamp','task','query','reference','challenge','topic','sub
22
  ABBREV_CHARS = 8
23
  ENTITY_CHOICES = ('identity', 'hotkey', 'coldkey')
24
  LOCAL_WANDB_PATH = './data/wandb'
25
- USERNAME = 'login19861986'
26
 
27
  # Initialize wandb with anonymous login
28
  wandb.login(anonymous='must')
@@ -139,8 +139,8 @@ def load_downloaded_runs(time, cols=KEYS):
139
  'question-answering': 'qa',
140
  }
141
 
142
- # # Runs which do not have a turn field are imputed to be turn zero (single turn)
143
- # df_all.turn.fillna(0, inplace=True)
144
 
145
  df_all.sort_values(by=['_timestamp'], inplace=True)
146
  # Check if df_all has the task columns
 
22
  ABBREV_CHARS = 8
23
  ENTITY_CHOICES = ('identity', 'hotkey', 'coldkey')
24
  LOCAL_WANDB_PATH = './data/wandb'
25
+ USERNAME = 'macrocrux'
26
 
27
  # Initialize wandb with anonymous login
28
  wandb.login(anonymous='must')
 
139
  'question-answering': 'qa',
140
  }
141
 
142
+ # Fill the missing values in the turn column with 0
143
+ df_all['turn'] = df_all['turn'].fillna(0)
144
 
145
  df_all.sort_values(by=['_timestamp'], inplace=True)
146
  # Check if df_all has the task columns