Testing whether timeout for taking too long works.

entrypoint.sh

@@ -19,9 +19,14 @@ python /parse_requests.py | while read -r line; do
     mkdir -p "$run_dir"
 
     # Let the benchmarking begin!
-    optimum-benchmark --config-name "${experiment_name}"  --config-dir /optimum-benchmark/examples/energy_star/ backend.model="${backend_model}" backend.processor="${backend_model}" hydra.run.dir="${run_dir}" 2> "${run_dir}/error.log" ||
+    timeout 5 optimum-benchmark --config-name "${experiment_name}"  --config-dir /optimum-benchmark/examples/energy_star/ backend.model="${backend_model}" backend.processor="${backend_model}" hydra.run.dir="${run_dir}" 2> "${run_dir}/error.log" ||
     # If the benchmark fails, update accordingly. Remove the run directory for that specific model.
     (python /failed_run.py --run_dir "${run_dir}" --model_name "${backend_model}" && rm -rf $run_dir)
+    if [ $? -eq 124 ]; then
+      # Timeout occurred
+      echo "timeout occurred"
+      python /failed_run.py --run_dir "${run_dir}" --model_name "${backend_model}" --reason "TAKES_TOO_LONG" && rm -rf $run_dir
+    fi
 done
 
 echo "Finished; uploading dataset results"