Changing names to AIEnergyScore

Dockerfile

@@ -76,4 +76,4 @@ RUN --mount=type=secret,id=DEBUG,mode=0444,required=true \
  git init && \
  git remote add origin $(cat /run/secrets/DEBUG)
 
-CMD ["/entrypoint.sh"]
+ENTRYPOINT ["/entrypoint.sh"]

create_results.py

@@ -10,28 +10,32 @@ api = HfApi(token=TOKEN)
 
 out_dir = sys.argv[1]
 
-#Uploading results
+# Uploading results
 api.upload_folder(
     folder_path=out_dir,
     repo_id="AIEnergyScore/results_debug",
     repo_type="dataset",
 )
 
-#Updating requests
-requests = load_dataset("AIEnergyScore/requests_debug", split="test", token=TOKEN)
+# Updating requests
+requests = load_dataset("AIEnergyScore/requests_debug", split="test",
+                        token=TOKEN)
 requests_dset = requests.to_pandas()
 
-models_ran=[]
+models_ran = []
 for f in os.scandir(out_dir):
-  if f.is_dir():
-    for s in os.scandir(f):
-      if s.is_dir() and s.name not in ['hooks','info','objects','refs','logs']:
-          for m in os.scandir(s):
-            models_ran.append(s.name+'/' + m.name)
-              
+    if f.is_dir():
+        for s in os.scandir(f):
+            if s.is_dir() and s.name not in ['hooks', 'info', 'objects', 'refs',
+                                             'logs']:
+                for m in os.scandir(s):
+                    models_ran.append(s.name + '/' + m.name)
+
 print("Models ran are: " + str(models_ran))
 
-requests_dset.loc[requests_dset["model"].isin(models_ran), ['status']] = "COMPLETED"
-updated_dset =Dataset.from_pandas(requests_dset)
-updated_dset.push_to_hub("AIEnergyScore/requests_debug", split="test", token=TOKEN)
+requests_dset.loc[
+    requests_dset["model"].isin(models_ran), ['status']] = "COMPLETED"
+updated_dset = Dataset.from_pandas(requests_dset)
+updated_dset.push_to_hub("AIEnergyScore/requests_debug", split="test",
+                         token=TOKEN)
 print("Updated model status")

entrypoint.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 
-export SPACE="EnergyStarAI/launch-computation-example"
+export SPACE="AIEnergyScore/launch-computation-example"
 
 echo "Not checking h100 -- already know it's not there."
 #python /check_h100.py
@@ -31,4 +31,4 @@ python /upload_run_folder.py --run_dir "/runs"
 # Pausing space
 echo "Pausing space."
 python /pause_space.py
-echo "Done."
+echo "Done."

failed_run.py

@@ -26,7 +26,7 @@ parser.add_argument(
 args = parser.parse_args()
 
 # Updating request
-dataset = load_dataset("EnergyStarAI/requests_debug", split="test",
+dataset = load_dataset("AIEnergyScore/requests_debug", split="test",
                        token=TOKEN).to_pandas()
 
 # Set benchmark to failed
@@ -48,5 +48,5 @@ except FileNotFoundError as e:
     print(f"Could not find {args.run_dir}/error.log")
 
 updated_dataset = Dataset.from_pandas(dataset)
-updated_dataset.push_to_hub("EnergyStarAI/requests_debug", split="test",
-                            token=TOKEN)
+updated_dataset.push_to_hub("AIEnergyScore/requests_debug", split="test",
+                            token=TOKEN)

parse_requests.py

@@ -1,10 +1,10 @@
 import os
 from datasets import load_dataset, Dataset
 
-skip_statuses = ['COMPLETED', 'FAILED', 'RUNNING']
 TOKEN = os.environ.get("DEBUG")
 requests_dataset = load_dataset("AIEnergyScore/requests_debug", split="test")
 
+
 def normalize_task(task):
     # Makes assumption about how the task names are being written, and called.
     return '_'.join(task.split()).lower()
@@ -12,9 +12,7 @@ def normalize_task(task):
 
 requests_dset = requests_dataset.to_pandas()
 
-for model, task in requests_dset[['model','task']].loc[requests_dset['status'] == 'PENDING'].to_dict(orient= 'split', index=False)['data']:
+for model, task in requests_dset[['model', 'task']].loc[
+    requests_dset['status'] == 'PENDING'].to_dict(orient='split', index=False)[
+    'data']:
     print("%s,%s" % (model, normalize_task(task)))
-
-# Custom errors we will rerun.
-for model, task in requests_dset[['model','task']].loc[~requests_dset['status'].isin(skip_statuses)].to_dict(orient= 'split', index=False)['data']:
-    print("%s,%s" % (model, normalize_task(task)))

pause_space.py

@@ -3,4 +3,4 @@ import os
 
 TOKEN = os.environ.get("DEBUG")
 API = HfApi(token=TOKEN)
-API.pause_space("EnergyStarAI/launch-computation-example")
+API.pause_space("AIEnergyScore/launch-computation-example")