bump version: 1.0.0

alibaba · Jun 29, 2020 · 1430719 · 1430719
1 parent aadb9d7
commit 1430719
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/packages/plugins/model-define/tensorflow-mobilenet-model-define/package.json b/packages/plugins/model-define/tensorflow-mobilenet-model-define/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@pipcook/plugins-tensorflow-mobilenet-model-define",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "mobilenet model of tensorflow",
   "main": "dist/index",
   "types": "dist/index",

diff --git a/packages/plugins/model-define/tensorflow-mobilenet-model-define/src/index.ts b/packages/plugins/model-define/tensorflow-mobilenet-model-define/src/index.ts
@@ -116,7 +116,7 @@ const mobilenetDefine: ModelDefineType = async (data: ImageDataset, args: ModelD
         channels: 3
       }));
       const shape = tf.shape(image).numpy();
-      return this.model.predict(tf.reshape(image, [ 1 ].concat(shape.slice(0, 3)))).toString();
+      return this.model.predict(tf.reshape(image, [ 1 ].concat(...shape.slice(0, 3)))).toString();
     }
   };
   return result;

diff --git a/packages/plugins/model-define/tensorflow-resnet-model-define/package.json b/packages/plugins/model-define/tensorflow-resnet-model-define/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@pipcook/plugins-tensorflow-resnet-model-define",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "resnet model of tensorflow",
   "main": "dist/index",
   "types": "dist/index",

diff --git a/packages/plugins/model-define/tensorflow-resnet-model-define/src/index.ts b/packages/plugins/model-define/tensorflow-resnet-model-define/src/index.ts
@@ -114,7 +114,7 @@ const resnetModelDefine: ModelDefineType = async (data: ImageDataset, args: Mode
         channels: 3
       }));
       const shape = tf.shape(image).numpy();
-      return this.model.predict(tf.reshape(image, [ 1 ].concat(shape.slice(0, 3)))).toString();
+      return this.model.predict(tf.reshape(image, [ 1 ].concat(...shape.slice(0, 3)))).toString();
     }
   };
   return result;