Address test failures.

DataDog · Apr 16, 2024 · a2576a4 · a2576a4
1 parent 98ca851
commit a2576a4
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -13,6 +13,7 @@ RUN pip install . -t ./python/lib/$runtime/site-packages
 
 # Remove *.pyc files
 RUN find ./python/lib/$runtime/site-packages -name \*.pyc -delete
+RUN find ./python/lib/$runtime/site-packages -name __pycache__ -delete
 
 # Remove botocore (40MB) to reduce package size. aws-xray-sdk
 # installs it, while it's already provided by the Lambda Runtime.

diff --git a/pyproject.toml b/pyproject.toml
@@ -33,9 +33,7 @@ urllib3 = [
     {version = "<2.0.0", python = "<3.11", optional = true},
     {version = "<2.1.0", python = ">=3.11", optional = true},
 ]
-importlib_metadata = {version = "*", python = "<3.8"}
 boto3 = { version = "^1.28.0", optional = true }
-typing_extensions = {version = "^4.0", python = "<3.8"}
 requests = { version ="^2.22.0", optional = true }
 pytest = { version= "^8.0.0", optional = true }
 pytest-benchmark = { version = "^4.0", optional = true }

diff --git a/tests/test_wrapper.py b/tests/test_wrapper.py
@@ -534,6 +534,7 @@ def lambda_handler(event, context):
         lambda_context = get_mock_context()
         test_span = tracer.trace("test_span")
         trace_ctx = tracer.current_trace_context()
+        trace_ctx.sampling_priority = 1
         test_span.finish()
         lambda_handler.inferred_span = test_span
         lambda_handler.make_inferred_span = False