Add --env flag to python client logs after deploy (#1487)

Author: deliahu

cli/cmd/root.go

@@ -213,9 +213,10 @@ func printEnvIfNotSpecified(envName string, cmd *cobra.Command) error {
 		return err
 	}
 
-	if _flagOutput == flags.PrettyOutputType {
+	if out != "" {
 		fmt.Print(out)
 	}
+
 	return nil
 }
 

pkg/workloads/cortex/client/cortex/client.py

@@ -83,14 +83,14 @@ def stream_to_stdout(process):
             env = os.environ.copy()
             env["CORTEX_CLI_INVOKER"] = "python"
             process = subprocess.Popen(
-                [get_cli_path(), "logs", api_name],
+                [get_cli_path(), "logs", "--env", self.env, api_name],
                 stderr=subprocess.STDOUT,
                 stdout=subprocess.PIPE,
                 encoding="utf8",
                 env=env,
             )
 
-            streamer = threading.Thread(target=stream_to_stdout, args=(process,))
+            streamer = threading.Thread(target=stream_to_stdout, args=[process])
             streamer.start()
 
             while process.poll() is None: