dn6 HF staff commited on
Commit
6860fc5
β€’
1 Parent(s): d662749

change how existing experiment is retrieved

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -53,7 +53,7 @@ def create_experiment(
53
  "api_key": comet_api_key,
54
  "workspace": comet_workspace,
55
  "project_name": comet_project_name,
56
- "previous_experiment": api_experiment.id,
57
  }
58
 
59
  return experiment, f"Started {api_experiment.name}. Happy logging!😊"
@@ -62,9 +62,16 @@ def create_experiment(
62
  return None, e
63
 
64
 
65
- def get_experiment(kwargs):
 
 
 
 
 
66
  try:
67
- return comet_ml.APIExperiment(**kwargs)
 
 
68
  except Exception as e:
69
  return None
70
 
@@ -138,7 +145,7 @@ def start_interface():
138
  )
139
 
140
  with gr.Row():
141
- start = gr.Button("Create Experiment", variant="primary")
142
  status = gr.Button("Experiment Status")
143
 
144
  status_output = gr.Markdown(label="Status")
 
53
  "api_key": comet_api_key,
54
  "workspace": comet_workspace,
55
  "project_name": comet_project_name,
56
+ "experiment_name": api_experiment.name,
57
  }
58
 
59
  return experiment, f"Started {api_experiment.name}. Happy logging!😊"
 
62
  return None, e
63
 
64
 
65
+ def get_experiment(experiment_state):
66
+ api_key = experiment_state["api_key"]
67
+ workspace = experiment_state["workspace"]
68
+ project = experiment_state["project_name"]
69
+ experiment_name = experiment_state["experiment_name"]
70
+
71
  try:
72
+ return comet_ml.API(api_key=api_key).get_experiment(
73
+ workspace=workspace, project=project, experiment_name=experiment_name
74
+ )
75
  except Exception as e:
76
  return None
77
 
 
145
  )
146
 
147
  with gr.Row():
148
+ start = gr.Button("Start Experiment", variant="primary")
149
  status = gr.Button("Experiment Status")
150
 
151
  status_output = gr.Markdown(label="Status")