chore: pass Labels/project/workspace to TaskSpec

determined-ai · Mar 8, 2023 · 0a034fd · 0a034fd
1 parent 7d8b711
commit 0a034fd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/master/internal/core_experiment.go b/master/internal/core_experiment.go
@@ -563,6 +563,12 @@ func (m *Master) parseCreateExperiment(params *CreateExperimentParams, user *mod
 		dbExp.Username = user.Username
 	}
 
+	taskSpec.Project = *config.RawProject
+	taskSpec.Workspace = *config.RawWorkspace
+	for label := range config.RawLabels {
+		taskSpec.Labels = append(taskSpec.Labels, label)
+	}
+
 	return dbExp, config, project, params.ValidateOnly, &taskSpec, err
 }
 

diff --git a/master/pkg/tasks/task.go b/master/pkg/tasks/task.go
@@ -83,6 +83,10 @@ type TaskSpec struct {
 	PbsConfig        expconf.PbsConfig
 
 	ExtraProxyPorts expconf.ProxyPortsConfig
+
+	Workspace string
+	Project   string
+	Labels    []string
 }
 
 // ResolveWorkDir resolves the work dir.