Move patch_all to init.

Author: purple4reina

datadog_lambda/__init__.py

@@ -17,3 +17,10 @@
 
 
 initialize_logging(__name__)
+
+
+from datadog_lambda.patch import patch_all  # noqa: E402
+
+# Patch third-party libraries for tracing, must be done before importing any
+# handler code.
+patch_all()

datadog_lambda/wrapper.py

@@ -25,7 +25,6 @@
     Headers,
 )
 from datadog_lambda.module_name import modify_module_name
-from datadog_lambda.patch import patch_all
 from datadog_lambda.span_pointers import calculate_span_pointers
 from datadog_lambda.tag_object import tag_object
 from datadog_lambda.tracing import (
@@ -47,10 +46,6 @@
     extract_http_status_code_tag,
 )
 
-# Patch third-party libraries for tracing, must be done before importing any
-# handler code.
-patch_all()
-
 if config.profiling_enabled:
     from ddtrace.profiling import profiler