black

Author: duncanista

datadog_lambda/tracing.py

@@ -387,6 +387,7 @@ def extract_context_from_kinesis_event(event, lambda_context):
 
 def _deterministic_sha256_hash(s: str, part: str) -> int:
     import hashlib
+
     sha256_hash = hashlib.sha256(s.encode()).hexdigest()
     # First two chars is '0b'. zfill to ensure 256 bits, but we only care about the first 128 bits
     binary_hash = bin(int(sha256_hash, 16))[2:].zfill(256)

datadog_lambda/wrapper.py

@@ -245,6 +245,7 @@ def __call__(self, event, context, **kwargs):
         except Exception:
             if not should_use_extension:
                 from datadog_lambda.metric import submit_invocations_metric
+
                 submit_invocations_metric(context)
 
             if self.span:
@@ -282,9 +283,10 @@ def _before(self, event, context):
         try:
             self.response = None
             set_cold_start(init_timestamp_ns)
-            
+
             if not should_use_extension:
                 from datadog_lambda.metric import submit_invocations_metric
+
                 submit_invocations_metric(context)
 
             self.trigger_tags = extract_trigger_tags(event, context)
@@ -386,6 +388,7 @@ def _after(self, event, context):
 
             if not self.flush_to_log or should_use_extension:
                 from datadog_lambda.metric import flush_stats
+
                 flush_stats(context)
             if should_use_extension and self.local_testing_mode:
                 # when testing locally, the extension does not know when an