Making this work with stable Covalent

.github/workflows/docker.yml

@@ -75,6 +75,7 @@ jobs:
         with:
           persist-credentials: false
           fetch-depth: 0
+          ref: post-qelectron-fixes
 
       - name: Setup QEMU
         uses: docker/setup-qemu-action@v2

.gitignore

@@ -12,4 +12,4 @@ scratch
 **/*.tfstate*
 **/.terraform.lock.hcl
 **/*.tfvars
-**/*.plan
+**/*.plan

covalent_awslambda_plugin/assets/infra/variables.tf

@@ -15,7 +15,7 @@
 # limitations under the License.
 
 variable "name" {
-  default = "covalent-lambda"
+  default = "sankalp-covalent-lambda"
   description = "Prefix to use for all provisioned resources"
 }
 

covalent_awslambda_plugin/assets/infra/versions.tf

@@ -18,7 +18,7 @@ terraform {
   required_providers {
     aws = {
       source  = "hashicorp/aws"
-      version = "5.17.0"
+      version = "~> 5.17.0"
     }
   }
 }

covalent_awslambda_plugin/exec.py

@@ -18,6 +18,9 @@
 
 import json
 import os
+import sys
+import traceback
+from unittest.mock import MagicMock
 
 import boto3
 import cloudpickle as pickle
@@ -40,6 +43,9 @@ def handler(event, context):
         s3 = boto3.client("s3")
         s3.download_file(s3_bucket, func_filename, local_func_filename)
 
+        # Patching mpire to avoid issues with multiprocessing on AWS Lambda
+        sys.modules["mpire"] = MagicMock()
+
         with open(local_func_filename, "rb") as f:
             function, args, kwargs = pickle.load(f)
 
@@ -49,8 +55,14 @@ def handler(event, context):
 
         s3.upload_file(local_result_filename, s3_bucket, result_filename)
     except Exception as ex:
+
+        # Capture the full traceback
+        tb = "".join(traceback.TracebackException.from_exception(ex).format())
+
         # Write json and upload to S3
         with open(local_exception_filename, "w") as f:
-            json.dump(str(ex), f)
+            json.dump(tb, f)
 
         s3.upload_file(local_exception_filename, s3_bucket, exception_filename)
+
+        raise ex

requirements.txt

@@ -1,2 +1,2 @@
-boto3==1.24.35
-covalent-aws-plugins>=0.12.0,<1
+boto3>=1.28.5
+covalent-aws-plugins>=0.19.0,<1