Read packages from environment variables (#1002)

Signed-off-by: Yee Hing Tong <[email protected]>
flyteorg · Jun 17, 2022 · 5bc9438 · 5bc9438
1 parent 9de00b4
commit 5bc9438
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flytekit/clis/sdk_in_container/pyflyte.py b/flytekit/clis/sdk_in_container/pyflyte.py
@@ -51,8 +51,8 @@ def main(ctx, pkgs=None, config=None):
     """
     ctx.obj = dict()
 
-    # Handle package management - get from config if not specified on the command line
-    pkgs = pkgs or []
+    # Handle package management - get from the command line, the environment variables, then the config file.
+    pkgs = pkgs or LocalSDK.WORKFLOW_PACKAGES.read() or []
     if config:
         ctx.obj[CTX_CONFIG_FILE] = config
         cfg = configuration.ConfigFile(config)