linting

Author: michael-zhao459

datadog_lambda/wrapper.py

@@ -247,7 +247,14 @@ def _before(self, event, context):
                         print(f"dd_json_data {dd_json_data}")
                         print(f"event_source {event_source}")
                         print(f"arn {arn}")
-                        carrier_get = lambda k: dd_json_data.get(k)
+
+                        def create_carrier_get(context_json):
+                            def carrier_get(key):
+                                return context_json.get(key)
+
+                            return carrier_get
+
+                        carrier_get = create_carrier_get(dd_json_data)
                         set_consume_checkpoint(event_source, arn, carrier_get)
 
                 set_dd_trace_py_root(trace_context_source, config.merge_xray_traces)

tests/test_tracing.py

@@ -42,9 +42,6 @@
     service_mapping as global_service_mapping,
     propagator,
     emit_telemetry_on_exception_outside_of_handler,
-    _create_carrier_get,
-    extract_context_from_sqs_or_sns_event_or_context,
-    extract_context_from_kinesis_event,
 )
 
 from tests.utils import get_mock_context