fix(format): remove check variables dir for savedmodel (#166)

kleveross · Dec 30, 2020 · cd8d716 · cd8d716
1 parent cdf84b0
commit cd8d716
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/pkg/model/format.go b/pkg/model/format.go
@@ -89,21 +89,14 @@ func ValidateError(modelPath string, modelName string, modelNum int32) error {
 
 func (f Format) validateForSavedModel(modelPath string, files []os.FileInfo) error {
 	var pbFileNum int32
-	var variablesDirNum int32
 	for _, file := range files {
 		if file.Name() == "saved_model.pb" {
 			pbFileNum++
 		}
-		if file.IsDir() && file.Name() == "variables" {
-			variablesDirNum++
-		}
 	}
 	if e := ValidateError(modelPath, "saved_model.pb", pbFileNum); e != nil {
 		return e
 	}
-	if e := ValidateError(modelPath, "variables", variablesDirNum); e != nil {
-		return e
-	}
 	return nil
 }