diff --git a/evalml/tests/utils_tests/test_gen_utils.py b/evalml/tests/utils_tests/test_gen_utils.py index 922b413c7b..51184e729d 100644 --- a/evalml/tests/utils_tests/test_gen_utils.py +++ b/evalml/tests/utils_tests/test_gen_utils.py @@ -37,14 +37,14 @@ def in_container_arm64(): The env var is set in the Dockerfile.arm for the purposes of local testing in a container on a mac M1, otherwise it's a noop. """ - if os.getenv("DOCKER_ARM", None): - import plotly.io as pio + # if os.getenv("DOCKER_ARM", None): + import plotly.io as pio - pio.kaleido.scope.chromium_args += ( - "--single-process", - "--headless", - "--disable-gpu", - ) + pio.kaleido.scope.chromium_args += ( + "--single-process", + "--headless", + "--disable-gpu", + ) @patch("importlib.import_module")