Add docker timeouts for tty

docker/compose#3927 docker/compose#3106 docker/compose#2338 docker/docker-py#630
astronomer · Sep 8, 2017 · 89e8674 · 89e8674
1 parent 2f57e6a
commit 89e8674
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/operators/docker_operator.py b/airflow/operators/docker_operator.py
@@ -150,7 +150,7 @@ def execute(self, context):
             )
             self.docker_url = self.docker_url.replace('tcp://', 'https://')
 
-        self.cli = APIClient(base_url=self.docker_url, version=self.api_version, tls=tls_config)
+        self.cli = APIClient(base_url=self.docker_url, version=self.api_version, tls=tls_config, timeout=120)
 
         if ':' not in self.image:
             image = self.image + ':latest'
@@ -186,6 +186,7 @@ def execute(self, context):
                     mem_limit=self.mem_limit,
                     user=self.user,
                     tty=True,
+                    stop_timeout=120,
             )
             self.cli.start(self.container['Id'])
             logging.info('started cli')