From 2f9f17a5c5993e65ee2b61d06f29ed5a6689d337 Mon Sep 17 00:00:00 2001
From: rashadmoarref <rashad.moarref@gmail.com>
Date: Thu, 2 Jun 2022 23:57:29 -0700
Subject: [PATCH] fix lambda deployment (#714)

* fixed lambda deployment

* Update aws_lambda.py

corrected typo in comment
---
 aws_lambda/aws_lambda.py | 4 ++++
 setup.py                 | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
index 780715a9..0b5ca884 100644
--- a/aws_lambda/aws_lambda.py
+++ b/aws_lambda/aws_lambda.py
@@ -704,6 +704,10 @@ def update_function(
             Publish=True,
         )
 
+    # Wait for function to be updated
+    waiter = client.get_waiter('function_updated')
+    waiter.wait(FunctionName=cfg.get("function_name"))
+
     kwargs = {
         "FunctionName": cfg.get("function_name"),
         "Role": role,
diff --git a/setup.py b/setup.py
index 5f6ca2f8..bce3297e 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
 
 REQUIREMENTS = [
     "boto3>=1.4.4",
-    "click==6.6",
+    "click>=6.6",
     "PyYAML==5.1",
 ]
 PACKAGE_DATA = {