Making model module arg optional

easy-model-fusion · Apr 8, 2024 · 400b8ea · 400b8ea
1 parent dee239f
commit 400b8ea
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/internal/downloader/model/utils.go b/internal/downloader/model/utils.go
@@ -66,12 +66,15 @@ func (a *Args) ToCobraTokenizer(cmd *cobra.Command) {
 func (a *Args) ToPython() []string {
 
 	// Mandatory arguments
-	cmdArgs := []string{TagPrefix + EmfClient, TagPrefix + Overwrite, a.DirectoryPath, a.ModelName, a.ModelModule}
+	cmdArgs := []string{TagPrefix + EmfClient, TagPrefix + Overwrite, a.DirectoryPath, a.ModelName}
 
 	// Optional arguments regarding the model
 	if a.ModelClass != "" {
 		cmdArgs = append(cmdArgs, TagPrefix+ModelClass, a.ModelClass)
 	}
+	if a.ModelModule != "" {
+		cmdArgs = append(cmdArgs, TagPrefix+ModelModule, a.ModelModule)
+	}
 	if len(a.ModelOptions) != 0 {
 		var options []string
 		for _, modelOption := range a.ModelOptions {