add default export

transcranial · Jan 2, 2018 · 82ec022 · 82ec022
1 parent 1088ba7
commit 82ec022
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "keras-js",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Run Keras models in the browser, with GPU support using WebGL",
   "main": "lib/index.js",
   "unpkg": "dist/keras.min.js",

diff --git a/src/index.js b/src/index.js
@@ -9,3 +9,6 @@ import * as testUtils from './utils/testUtils'
 const GPU_SUPPORT = webgl2.isSupported
 
 export { Model, Tensor, GPU_SUPPORT, activations, layers, testUtils }
+
+const KerasJS = { Model, Tensor, GPU_SUPPORT, activations, layers, testUtils }
+export { KerasJS as default }