Add deployment: input to pex deploy workflow

actions/build_deploy_python_executable/action.yml

@@ -8,6 +8,10 @@ inputs:
     description: 'Python version string, major.minor only, eg "3.8"'
     required: false
     default: '3.8'
+  deployment:
+    required: false
+    description: "The deployment to push to, defaults to 'prod'. Ignored for pull requests where the branch deployment is used."
+    default: "prod"  
   deploy:
     description: 'Whether to upload the code files and update the code location'
     required: false

src/deploy_pex.py

@@ -34,7 +34,8 @@ def main():
         deployment_name = get_branch_deployment_name(project_dir)
     else:
         print("Going to do a full deployment.", flush=True)
-        deployment_name = os.getenv("DAGSTER_CLOUD_DEPLOYMENT", "prod")
+        # INPUT_DEPLOYMENT is automatically set by github to the `deployment:` input value, if provided
+        deployment_name = os.getenv("INPUT_DEPLOYMENT", "prod")
 
     ubuntu_version = get_runner_ubuntu_version()
     print("Running on Ubuntu", ubuntu_version, flush=True)