reset self.response and lint

Author: kimi-p

datadog_lambda/constants.py

@@ -51,4 +51,4 @@ class XrayDaemon(object):
     "503": "503 Service Unavailable",
     "504": "504 Gateway Timeout",
     "505": "505 HTTP Version Not Supported",
-}
+}

datadog_lambda/wrapper.py

@@ -153,6 +153,7 @@ def __call__(self, event, context, **kwargs):
             self._after(event, context)
 
     def _before(self, event, context):
+        self.response = None
         try:
 
             set_cold_start()
@@ -198,11 +199,14 @@ def _after(self, event, context):
                     if self.span:
                         self.span.set_traceback()
                         self.span.error = 1
-                        self.span.set_tags({
-                            ERROR_TYPE: "5xx Server Errors",
-                            ERROR_MSG: SERVER_ERRORS_STATUS_CODES.get(status_code,
-                                                                      "5xx Server Errors"),
-                        })
+                        self.span.set_tags(
+                            {
+                                ERROR_TYPE: "5xx Server Errors",
+                                ERROR_MSG: SERVER_ERRORS_STATUS_CODES.get(
+                                    status_code,
+                                    "5xx Server Errors"),
+                            }
+                        )
             # Create a new dummy Datadog subsegment for function trigger tags so we
             # can attach them to X-Ray spans when hybrid tracing is used
             if self.trigger_tags: