Container request priority flag.

[lightning.git] / pca.go

diff --git a/pca.go b/pca.go
index cd68eeb897e7fefda071f97c1d2eb26142de12ad..34655d975384bf1a170b6a4bc6c6aca1fc977af0 100644 (file)
--- a/pca.go
+++ b/pca.go
@@ -22,6 +22,7 @@ func (cmd *pythonPCA) RunCommand(prog string, args []string, stdin io.Reader, st
        flags.SetOutput(stderr)
        projectUUID := flags.String("project", "", "project `UUID` for output data")
        inputFilename := flags.String("i", "-", "input `file`")
+       priority := flags.Int("priority", 500, "container request priority")
        err = flags.Parse(args)
        if err == flag.ErrHelp {
                err = nil
@@ -36,6 +37,7 @@ func (cmd *pythonPCA) RunCommand(prog string, args []string, stdin io.Reader, st
                ProjectUUID: *projectUUID,
                RAM:         120000000000,
                VCPUs:       1,
+               Priority:    *priority,
        }
        err = runner.TranslatePaths(inputFilename)
        if err != nil {
@@ -46,9 +48,11 @@ func (cmd *pythonPCA) RunCommand(prog string, args []string, stdin io.Reader, st
 import scipy
 from sklearn.decomposition import PCA
 scipy.save(sys.argv[2], PCA(n_components=4).fit_transform(scipy.load(sys.argv[1])))`, *inputFilename, "/mnt/output/pca.npy"}
-       err = runner.Run()
+       var output string
+       output, err = runner.Run()
        if err != nil {
                return 1
        }
+       fmt.Fprintln(stdout, output+"/pca.npy")
        return 0
 }