fix: test

aws · Jan 14, 2025 · e03d375 · e03d375
1 parent 4930607
commit e03d375
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py
@@ -225,7 +225,7 @@ def _validate_docker_images(
     # A consequence of this design decision is that any test assertions should go inside the container's entry-point.
 
     # Special handling for JupyterLab entrypoint testing
-    if dockerfile_path in ["recovery-mode.test.Dockerfile"]:
+    if dockerfile_path in ["sagemaker-recovery-mode.test.Dockerfile"]:
         _test_jupyterlab_entrypoint(image)
     else:
         container = _docker_client.containers.run(