Gil-Simas commited on
Commit
c29fdef
·
1 Parent(s): 76d1d0d

run name as input

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +5 -5
user-friendly-metrics.py CHANGED
@@ -131,7 +131,7 @@ class UserFriendlyMetrics(evaluate.Metric):
131
  formatted_datetime = current_datetime.strftime("%Y-%m-%d_%H-%M-%S")
132
  wandb.login(key=os.getenv("WANDB_API_KEY"))
133
 
134
- if not wandb_runs is None:
135
  assert len(wandb_runs) == len(results), "runs and results must have the same length"
136
  else:
137
  wandb_runs = [f"{i}_{formatted_datetime}" for i in list(results.keys())]
@@ -145,10 +145,10 @@ class UserFriendlyMetrics(evaluate.Metric):
145
  def wandb_run(self, result, wandb_run_name, wandb_project, debug):
146
 
147
  run = wandb.init(
148
- project=wandb_project,
149
- name=f"evaluation-{formatted_datetime}",
150
- reinit=True,
151
- settings=wandb.Settings(silent=not debug),
152
  )
153
 
154
  categories = {
 
131
  formatted_datetime = current_datetime.strftime("%Y-%m-%d_%H-%M-%S")
132
  wandb.login(key=os.getenv("WANDB_API_KEY"))
133
 
134
+ if wandb_runs is not None:
135
  assert len(wandb_runs) == len(results), "runs and results must have the same length"
136
  else:
137
  wandb_runs = [f"{i}_{formatted_datetime}" for i in list(results.keys())]
 
145
  def wandb_run(self, result, wandb_run_name, wandb_project, debug):
146
 
147
  run = wandb.init(
148
+ project = wandb_project,
149
+ name = wandb_run_name,
150
+ reinit = True,
151
+ settings = wandb.Settings(silent=not debug),
152
  )
153
 
154
  categories = {