clean

Author: michael-zhao459

datadog_lambda/tracing.py

@@ -383,8 +383,8 @@ def extract_context_from_kinesis_event(event, lambda_context):
             data_str = str_bytes.decode("ascii")
             data_obj = json.loads(data_str)
             dd_ctx = data_obj.get("_datadog")
-        if dd_ctx:
-            return propagator.extract(dd_ctx), dd_ctx, arn
+            if dd_ctx:
+                return propagator.extract(dd_ctx), dd_ctx, arn
     except Exception as e:
         logger.debug("The trace extractor returned with error %s", e)
 

datadog_lambda/wrapper.py

@@ -374,7 +374,7 @@ def carrier_get(key):
             return carrier_get
 
         carrier_get = create_carrier_get(dd_json_data)
-        set_consume_checkpoint(event_source, arn, carrier_get)
+        set_consume_checkpoint(event_source, arn, carrier_get, manual_checkpoint=False)
     except Exception as e:
         logger.debug(f"Failed to set DSM checkpoint: {e}")