Move patch_all to init.

Author: purple4reina

datadog_lambda/__init__.py

@@ -10,10 +10,17 @@
 
 initialize_cold_start_tracing()
 
+
 # The minor version corresponds to the Lambda layer version.
 # E.g.,, version 0.5.0 gets packaged into layer version 5.
 from datadog_lambda.version import __version__  # noqa: E402 F401
 from datadog_lambda.logger import initialize_logging  # noqa: E402
 
-
 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