diff --git a/iterative/resource_runner.go b/iterative/resource_runner.go index 03691a48..a3c12618 100644 --- a/iterative/resource_runner.go +++ b/iterative/resource_runner.go @@ -341,7 +341,7 @@ export KUBERNETES_CONFIGURATION={{escape .KUBERNETES_CONFIGURATION}} {{.runner_startup_script}} {{- end}} -HOME="$(mktemp -d)" exec $(which cml-runner || echo "cml runner") \ +HOME="$(mktemp -d)" exec $(which cml-runner || echo $(which cml-internal || echo cml) runner) \ {{if .name}} --name {{escape .name}}{{end}} \ {{if .labels}} --labels {{escape .labels}}{{end}} \ {{if .idle_timeout}} --idle-timeout {{escape .idle_timeout}}{{end}} \ diff --git a/iterative/testdata/script_template_cloud_aws.golden b/iterative/testdata/script_template_cloud_aws.golden index 1b5c846f..c16cf613 100644 --- a/iterative/testdata/script_template_cloud_aws.golden +++ b/iterative/testdata/script_template_cloud_aws.golden @@ -44,7 +44,7 @@ export AWS_SECRET_ACCESS_KEY='0 value with "quotes" and spaces' export AWS_ACCESS_KEY_ID='1 value with "quotes" and spaces' export AWS_SESSION_TOKEN='2 value with "quotes" and spaces' -HOME="$(mktemp -d)" exec $(which cml-runner || echo "cml runner") \ +HOME="$(mktemp -d)" exec $(which cml-runner || echo $(which cml-internal || echo cml) runner) \ --name '10 value with "quotes" and spaces' \ --labels '16 value with "quotes" and spaces' \ --idle-timeout 11 \ diff --git a/iterative/testdata/script_template_cloud_azure.golden b/iterative/testdata/script_template_cloud_azure.golden index b1564e0d..f83cefcd 100644 --- a/iterative/testdata/script_template_cloud_azure.golden +++ b/iterative/testdata/script_template_cloud_azure.golden @@ -45,7 +45,7 @@ export AZURE_CLIENT_SECRET='4 value with "quotes" and spaces' export AZURE_SUBSCRIPTION_ID='5 value with "quotes" and spaces' export AZURE_TENANT_ID='6 value with "quotes" and spaces' -HOME="$(mktemp -d)" exec $(which cml-runner || echo "cml runner") \ +HOME="$(mktemp -d)" exec $(which cml-runner || echo $(which cml-internal || echo cml) runner) \ --name '10 value with "quotes" and spaces' \ --labels '16 value with "quotes" and spaces' \ --idle-timeout 11 \ diff --git a/iterative/testdata/script_template_cloud_gcp.golden b/iterative/testdata/script_template_cloud_gcp.golden index 40fc7fe4..8372320f 100644 --- a/iterative/testdata/script_template_cloud_gcp.golden +++ b/iterative/testdata/script_template_cloud_gcp.golden @@ -42,7 +42,7 @@ sudo tee /usr/bin/cml.sh << 'EOF' #!/bin/sh export GOOGLE_APPLICATION_CREDENTIALS_DATA='7 value with "quotes" and spaces' -HOME="$(mktemp -d)" exec $(which cml-runner || echo "cml runner") \ +HOME="$(mktemp -d)" exec $(which cml-runner || echo $(which cml-internal || echo cml) runner) \ --name '10 value with "quotes" and spaces' \ --labels '16 value with "quotes" and spaces' \ --idle-timeout 11 \ diff --git a/iterative/testdata/script_template_cloud_invalid.golden b/iterative/testdata/script_template_cloud_invalid.golden index ff8c3a67..81c9683a 100644 --- a/iterative/testdata/script_template_cloud_invalid.golden +++ b/iterative/testdata/script_template_cloud_invalid.golden @@ -41,7 +41,7 @@ sudo npm config set user 0 && sudo npm install --global 18 value with "quotes" a sudo tee /usr/bin/cml.sh << 'EOF' #!/bin/sh -HOME="$(mktemp -d)" exec $(which cml-runner || echo "cml runner") \ +HOME="$(mktemp -d)" exec $(which cml-runner || echo $(which cml-internal || echo cml) runner) \ --name '10 value with "quotes" and spaces' \ --labels '16 value with "quotes" and spaces' \ --idle-timeout 11 \ diff --git a/iterative/testdata/script_template_cloud_kubernetes.golden b/iterative/testdata/script_template_cloud_kubernetes.golden index c557f2a5..ab7f31d0 100644 --- a/iterative/testdata/script_template_cloud_kubernetes.golden +++ b/iterative/testdata/script_template_cloud_kubernetes.golden @@ -2,7 +2,7 @@ sudo systemctl is-enabled cml.service && return 0 export KUBERNETES_CONFIGURATION='8 value with "quotes" and spaces' -HOME="$(mktemp -d)" exec $(which cml-runner || echo "cml runner") \ +HOME="$(mktemp -d)" exec $(which cml-runner || echo $(which cml-internal || echo cml) runner) \ --name '10 value with "quotes" and spaces' \ --labels '16 value with "quotes" and spaces' \ --idle-timeout 11 \ diff --git a/iterative/utils/helpers.go b/iterative/utils/helpers.go index 42092bc6..28ba71fb 100644 --- a/iterative/utils/helpers.go +++ b/iterative/utils/helpers.go @@ -44,7 +44,8 @@ func getGHCML(v string) string { return fmt.Sprintf(`sudo mkdir -p /opt/cml/ sudo curl --location --url %s --output /opt/cml/cml-linux sudo chmod +x /opt/cml/cml-linux -sudo ln -s /opt/cml/cml-linux /usr/bin/cml`, v) +sudo ln -s /opt/cml/cml-linux /usr/bin/cml +sudo ln /opt/cml/cml-linux /usr/bin/cml-internal`, v) // hard link to fix cml#920 } func getNPMCML(v string) string {