fix(CLI): Fix build output argument and huggingface integration (#886)

fix: Fix build output argument and huggingface integration Signed-off-by: Ce Gao <cegao@tensorchord.ai> Signed-off-by: Ce Gao <cegao@tensorchord.ai>
tensorchord · Sep 14, 2022 · 128f866 · 128f866
1 parent 2e8b5d5
commit 128f866
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/app/build.go b/pkg/app/build.go
@@ -176,7 +176,7 @@ func ParseBuildOpt(clicontext *cli.Context) (builder.Options, error) {
 	if err != nil {
 		return builder.Options{}, err
 	}
-	output := ""
+	output := clicontext.String("output")
 	exportCache := clicontext.String("export-cache")
 	importCache := clicontext.String("import-cache")
 	useProxy := clicontext.Bool("use-proxy")

diff --git a/pkg/lang/frontend/starlark/data/const.go b/pkg/lang/frontend/starlark/data/const.go
@@ -16,6 +16,6 @@ package data
 
 const (
 	ruleEnvdManagedDataSource = "data.envd"
-	huggingFaceDatasetPath    = "~/.cache/huggingface/datasets"
+	huggingFaceDatasetPath    = "~/.cache/huggingface"
 	dglFaceDatasetPath        = "~/.dgl"
 )