fix

Author: michael-zhao459

datadog_lambda/wrapper.py

@@ -9,8 +9,6 @@
 from importlib import import_module
 from time import time_ns
 
-from datadog_lambda.asm import asm_set_context, asm_start_response, asm_start_request
-from datadog_lambda.dsm import set_dsm_context
 from ddtrace.internal._exceptions import BlockingException
 from datadog_lambda.extension import should_use_extension, flush_extension
 from datadog_lambda.cold_start import (
@@ -259,6 +257,7 @@ def _before(self, event, context):
                     asm_set_context(event_source)
 
                 if config.data_streams_enabled:
+                    from datadog_lambda.dsm import set_dsm_context
                     set_dsm_context(event, event_source)
 
                 self.span = create_function_execution_span(