diff --git a/scripts/tests/pipeline_generator/test_plugin.py b/scripts/tests/pipeline_generator/test_plugin.py index 7e85ac0..4ced8ab 100644 --- a/scripts/tests/pipeline_generator/test_plugin.py +++ b/scripts/tests/pipeline_generator/test_plugin.py @@ -11,7 +11,7 @@ def test_get_kubernetes_plugin_config(): docker_image_path = "test_image:latest" - test_bash_command = ["echo", "Hello, Kubernetes!"] + test_bash_command = ["bash", "-c", "echo A"] num_gpus = 1 expected_config = { @@ -20,7 +20,7 @@ def test_get_kubernetes_plugin_config(): "containers": [ { "image": docker_image_path, - "command": [" ".join(test_bash_command)], + "command": ['bash -c "echo A"'], "resources": {"limits": {"nvidia.com/gpu": num_gpus}}, "volumeMounts": [ {"name": "devshm", "mountPath": "/dev/shm"},