From bd121d716996510d5a13c1b8a7b86d01cc06d0c0 Mon Sep 17 00:00:00 2001
From: znotdead <zhirafchik@gmail.com>
Date: Tue, 19 Apr 2022 12:06:58 -0700
Subject: [PATCH] 711 - publish new function after all changes done

---
 aws_lambda/aws_lambda.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
index 780715a9..e02bbf66 100644
--- a/aws_lambda/aws_lambda.py
+++ b/aws_lambda/aws_lambda.py
@@ -16,7 +16,6 @@
 import boto3
 import botocore
 import yaml
-import sys
 
 from .helpers import archive
 from .helpers import get_environment_variable_value
@@ -703,6 +702,10 @@ def update_function(
             ZipFile=byte_stream,
             Publish=True,
         )
+    waiter = client.get_waiter('function_updated')
+    waiter.wait(
+        FunctionName=cfg.get("function_name"),
+    )
 
     kwargs = {
         "FunctionName": cfg.get("function_name"),