reformat

GoogleCloudPlatform · Dec 6, 2024 · 5bdcef5 · 5bdcef5
1 parent 970b9ca
commit 5bdcef5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dags/map_reproducibility/nemo_gpt3.py b/dags/map_reproducibility/nemo_gpt3.py
@@ -87,10 +87,12 @@ def run_aotc_workload():
 
   # Extract COMPLETE_JOB_NAME from the output
   bucket_name, file_name = extract_bucket_file_name(result.output)
+  get_metrics_from_gcs(bucket_name, file_name)
+
+  # Extract PYTHONPATH from the output
   python_path = extract_python_path(result.output)
   sys.path.append(python_path)
 
-  get_metrics_from_gcs(bucket_name, file_name)
 
 with models.DAG(
     dag_id="reproducibility_nemo_gpt3_nighly_dag",