sasha HF staff commited on
Commit
211db76
1 Parent(s): 9fe35b9

Update create_results.py

Browse files
Files changed (1) hide show
  1. create_results.py +8 -1
create_results.py CHANGED
@@ -21,7 +21,14 @@ api.upload_folder(
21
  requests = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
22
  requests_dset = requests.to_pandas()
23
 
24
- models_ran=[ f.path for f in os.scandir(out_dir) if f.is_dir() ]
 
 
 
 
 
 
 
25
  print("Models ran are: " + str(models_ran))
26
 
27
  requests_dset.loc[requests_dset["model"].isin(models_ran), ['status']] = "COMPLETED"
 
21
  requests = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
22
  requests_dset = requests.to_pandas()
23
 
24
+ models_ran=[]
25
+ for f in os.scandir(out_dir):
26
+ if f.is_dir():
27
+ for s in os.scandir(f):
28
+ if s.is_dir() and s.name not in ['hooks','info','objects','refs','logs']:
29
+ for m in os.scandir(s):
30
+ models_ran.append(s.name+'/' + m.name)
31
+
32
  print("Models ran are: " + str(models_ran))
33
 
34
  requests_dset.loc[requests_dset["model"].isin(models_ran), ['status']] = "COMPLETED"