The workaround to fix the label loss (#741)

* The workaround to fix the label loss Signed-off-by: nullday <aseaday@hotmail.com> * Add comment Signed-off-by: nullday <aseaday@hotmail.com>
tensorchord · Aug 9, 2022 · 4df956b · 4df956b
1 parent dd609da
commit 4df956b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/builder/builder.go b/pkg/builder/builder.go
@@ -140,6 +140,11 @@ func (b generalBuilder) NumGPUs() int {
 
 func (b generalBuilder) Build(ctx context.Context, force bool) error {
  if !force && !b.checkIfNeedBuild(ctx) {
+ // The container label needs the interpreted defaultGrpah to be set in `StartEnvd`
+ // TODO(Qi Chen): remove this hack
+ if err := b.Interpret(); err != nil {
+ return errors.Wrap(err, "failed to interpret")
+ }
  return nil
  }