Update create_results.py
Browse files- 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=[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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"
|