From 00dddeb07727b80ed9dc214eaceb1d4cca206ae8 Mon Sep 17 00:00:00 2001 From: Devlin Date: Tue, 28 Jan 2025 19:49:54 -0800 Subject: [PATCH] rename `INPUT_TEXT` environment variable to `INPUT` --- src/create_lilypad_module/templates/lilypad_module.json.tmpl | 2 +- src/create_lilypad_module/templates/scripts/run_module.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/create_lilypad_module/templates/lilypad_module.json.tmpl b/src/create_lilypad_module/templates/lilypad_module.json.tmpl index e821518..0539ed0 100644 --- a/src/create_lilypad_module/templates/lilypad_module.json.tmpl +++ b/src/create_lilypad_module/templates/lilypad_module.json.tmpl @@ -16,7 +16,7 @@ "Entrypoint": ["python", "/workspace/src/run_inference.py"], "WorkingDirectory": "/workspace", "EnvironmentVariables": [ - "INPUT_TEXT={{ js .input }}" + "INPUT={{ js .input }}" ], "Image": "", }, diff --git a/src/create_lilypad_module/templates/scripts/run_module.py b/src/create_lilypad_module/templates/scripts/run_module.py index 38b655c..ed4a352 100644 --- a/src/create_lilypad_module/templates/scripts/run_module.py +++ b/src/create_lilypad_module/templates/scripts/run_module.py @@ -32,7 +32,7 @@ def run_module(): "docker", "run", "-e", - f"INPUT_TEXT={input}", + f"INPUT={input}", "-v", "$(pwd)/outputs:/outputs", f"{DOCKER_REPO}:latest",