fix bugs in run.py

airbnb · Feb 28, 2024 · f20ee00 · f20ee00
1 parent 992a0e3
commit f20ee00
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/api/py/ai/chronon/repo/run.py b/api/py/ai/chronon/repo/run.py
@@ -276,7 +276,10 @@ def set_runtime_env(args):
                     team = "default"
                 # context is the environment in which the job is running, which is provided from the args,
                 # default to be dev.
-                context = args.env
+                if args.env:
+                    context = args.env
+                else:
+                    context = "dev"
                 logging.info(
                     f"Context: {context} -- conf_type: {conf_type} -- team: {team}"
                 )