From 07ba81011081dfd3d922798fdae3d2610f71d678 Mon Sep 17 00:00:00 2001
From: Daniel Barnes <dabarnes2b@gmail.com>
Date: Wed, 15 Mar 2023 01:15:55 +0900
Subject: [PATCH] patch: use same env as dvclive (#745)

* use same env as dvclive

* go test ./... -update
---
 task/common/machine/machine-script.sh.tpl                  | 2 +-
 task/common/machine/testdata/machine_script_full.golden    | 2 +-
 task/common/machine/testdata/machine_script_minimal.golden | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/task/common/machine/machine-script.sh.tpl b/task/common/machine/machine-script.sh.tpl
index 6aff8cee..c126fcb5 100755
--- a/task/common/machine/machine-script.sh.tpl
+++ b/task/common/machine/machine-script.sh.tpl
@@ -17,7 +17,7 @@ chmod u=rwx,g=rx,o=rx /usr/bin/tpi-task-shutdown
 
 sudo tee /usr/bin/tpi-task-studio-log << 'END'
 #!/bin/bash
-URL="${STUDIO_URL:-https://studio.iterative.ai/api/live}"
+URL="${STUDIO_ENDPOINT:-https://studio.iterative.ai/api/live}"
 STEP="${STUDIO_STEP:-`echo $(date +%s)`}"
 STATUS=$1
 DATE_START="${TPI_TASK_DATE_START:-0}"
diff --git a/task/common/machine/testdata/machine_script_full.golden b/task/common/machine/testdata/machine_script_full.golden
index 47997445..7b9ec37b 100644
--- a/task/common/machine/testdata/machine_script_full.golden
+++ b/task/common/machine/testdata/machine_script_full.golden
@@ -17,7 +17,7 @@ chmod u=rwx,g=rx,o=rx /usr/bin/tpi-task-shutdown
 
 sudo tee /usr/bin/tpi-task-studio-log << 'END'
 #!/bin/bash
-URL="${STUDIO_URL:-https://studio.iterative.ai/api/live}"
+URL="${STUDIO_ENDPOINT:-https://studio.iterative.ai/api/live}"
 STEP="${STUDIO_STEP:-`echo $(date +%s)`}"
 STATUS=$1
 DATE_START="${TPI_TASK_DATE_START:-0}"
diff --git a/task/common/machine/testdata/machine_script_minimal.golden b/task/common/machine/testdata/machine_script_minimal.golden
index 913a8ef0..56811200 100644
--- a/task/common/machine/testdata/machine_script_minimal.golden
+++ b/task/common/machine/testdata/machine_script_minimal.golden
@@ -17,7 +17,7 @@ chmod u=rwx,g=rx,o=rx /usr/bin/tpi-task-shutdown
 
 sudo tee /usr/bin/tpi-task-studio-log << 'END'
 #!/bin/bash
-URL="${STUDIO_URL:-https://studio.iterative.ai/api/live}"
+URL="${STUDIO_ENDPOINT:-https://studio.iterative.ai/api/live}"
 STEP="${STUDIO_STEP:-`echo $(date +%s)`}"
 STATUS=$1
 DATE_START="${TPI_TASK_DATE_START:-0}"