From 9a987d7bbea08e2348dd236f21030e8e1d52699d Mon Sep 17 00:00:00 2001
From: Cunliang Kong <30660718+styxjedi@users.noreply.github.com>
Date: Thu, 6 Jan 2022 00:00:00 +0800
Subject: [PATCH] Update execution.py

when executing on a local cluster the timeout is not being set.
---
 cc_net/execution.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cc_net/execution.py b/cc_net/execution.py
index d7664d2..67e5745 100644
--- a/cc_net/execution.py
+++ b/cc_net/execution.py
@@ -55,6 +55,7 @@ def get_executor(
     cluster = None if execution_mode == "auto" else execution_mode
     # use submitit to detect which executor is available
     ex = submitit.AutoExecutor(log_dir, cluster=cluster)
+    ex.parameters['timeout_min'] = int(timeout_hour * 60)
 
     if ex.cluster == "local":
         # LocalExecutor doesn't respect task_parallelism