Spaces:
Sleeping
Sleeping
bkb2135
commited on
Commit
·
1adf887
1
Parent(s):
be32e3b
Use macrocrux and fill turn
Browse files
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 = '
|
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 |
-
#
|
143 |
-
|
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
|