Revert "test 3"

This reverts commit de35257.
alteryx · May 21, 2024 · abaf00d · abaf00d
1 parent de35257
commit abaf00d
Showing 1 changed file with 7 additions and 7 deletions.
diff --git 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")