fix DeepSource issue: Function literal can be simplified

Signed-off-by: datelier <57349093+datelier@users.noreply.github.com>
vdaas · Jun 3, 2020 · 057e0bd · 057e0bd
1 parent 0df8eb0
commit 057e0bd
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/internal/core/converter/tensorflow/tensorflow.go b/internal/core/converter/tensorflow/tensorflow.go
@@ -69,9 +69,7 @@ const (
 	threeDim
 )
 
-var loadFunc = func(exportDir string, tags []string, options *SessionOptions) (*tf.SavedModel, error) {
-	return tf.LoadSavedModel(exportDir, tags, options)
-}
+var loadFunc = tf.LoadSavedModel
 
 // New load a tensorlfow model and returns a new tensorflow struct.
 func New(opts ...Option) (TF, error) {