diff --git a/datadog_lambda/wrapper.py b/datadog_lambda/wrapper.py index 99678ffa..f4230372 100644 --- a/datadog_lambda/wrapper.py +++ b/datadog_lambda/wrapper.py @@ -59,7 +59,6 @@ def __init__(self, func): def _before(self, event, context): set_cold_start() - try: submit_invocations_metric(context) # Extract Datadog trace context from incoming requests @@ -77,10 +76,10 @@ def _after(self, event, context): except Exception: traceback.print_exc() - def __call__(self, event, context): + def __call__(self, event, context, **kwargs): self._before(event, context) try: - return self.func(event, context) + return self.func(event, context, **kwargs) except Exception: submit_errors_metric(context) raise diff --git a/scripts/run_tests.sh b/scripts/run_tests.sh index c8643de7..5edccd90 100755 --- a/scripts/run_tests.sh +++ b/scripts/run_tests.sh @@ -1,4 +1,4 @@ -#!/bin/sh +#!/bin/bash # Unless explicitly stated otherwise all files in this repository are licensed # under the Apache License Version 2.0.