diff --git a/python/aws_lambda_powertools/logging/logger.py b/python/aws_lambda_powertools/logging/logger.py index 0e9b22407d2..4b9a232bad9 100644 --- a/python/aws_lambda_powertools/logging/logger.py +++ b/python/aws_lambda_powertools/logging/logger.py @@ -5,8 +5,7 @@ from distutils.util import strtobool from typing import Any, Callable, Dict -import aws_lambda_logging - +from . import aws_lambda_logging from ..helper.models import MetricUnit, build_lambda_context_model, build_metric_unit_from_str logger = logging.getLogger(__name__) diff --git a/python/tests/functional/test_aws_lambda_logging.py b/python/tests/functional/test_aws_lambda_logging.py index cdefb9f1171..79251a25274 100644 --- a/python/tests/functional/test_aws_lambda_logging.py +++ b/python/tests/functional/test_aws_lambda_logging.py @@ -112,25 +112,4 @@ class X: log_dict = json.loads(stdout.getvalue()) - assert log_dict["message"]["x"].startswith("<") - - -def test_wrap(): - from aws_lambda_logging import wrap - - @wrap - def handler(event, context): - pass - - handler(None, None) - - -def test_wrap_when_request_id_is_in_event(): - from aws_lambda_logging import wrap - - @wrap - def handler(event, context): - pass - - event = {"requestContext": {"requestId": "c19e27e6-230b-11e8-ba47-a95ceb1c6480"}} - handler(event, None) + assert log_dict["message"]["x"].startswith("<") \ No newline at end of file