Issue #5 fix bug in handle_infractions in put_events

VerticalRelevance · May 3, 2022 · e045945 · e045945
1 parent d65dd4c
commit e045945
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/supplementary_files/lambdas/handle_infraction/lambda_function.py b/supplementary_files/lambdas/handle_infraction/lambda_function.py
@@ -8,9 +8,9 @@
 eb = boto3.client('events')
 
 def update_item(*,
-    table,
-    pk,
-    sk,
+    table:str,
+    pk:str,
+    sk:str,
     attributes:dict[str,str]
 ):
 
@@ -56,8 +56,8 @@ def ddb_compatible_type(Item):
         return True
 
 def put_event_entry(*,
-    event_bus_name,
-    source,
+    event_bus_name:str,
+    source:str,
     detail:dict
 ):
     try:
@@ -67,7 +67,7 @@ def put_event_entry(*,
                     'EventBusName':event_bus_name,
                     'Detail':json.dumps(detail),
                     'DetailType':os.environ.get('AWS_LAMBDA_FUNCTION_NAME'),
-                    'source':source,
+                    'Source':source,
                 }
             ]
         )