Updated scripts for distribution.

package.json

@@ -3,7 +3,8 @@
   "version": "0.0.1",
   "private": false,
   "description": "A library using TensorFlow.js for Deep Reinforcement Learning",
-  "main": "./dist/furnish.js",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/Pravez/FurnishJS.git"
@@ -44,7 +45,8 @@
     "sinon-chai": "^3.0.0",
     "ts-node": "^6.0.3",
     "tsify": "^4.0.0",
-    "typescript": "^2.8.3"
+    "typescript": "^2.8.3",
+    "uglify-js": "^3.3.27"
   },
   "files": [
     "dist/furnish.js"

scripts/build-npm.sh

@@ -4,6 +4,6 @@ set -e
 
 rm -rf dist/
 tsc -p .
-browserify --standalone Furnish src/furnish.ts -p [tsify] > dist/furnish.js
+browserify --standalone Furnish src/furnish.ts -p [tsify] | uglifyjs > dist/furnish.js
 echo "Prepared bundle"
 npm pack

src/furnish/model.ts

@@ -62,7 +62,7 @@ export class Model {
         return new Result(<Tensor> this.model.predict(x, config));
     }
 
-    fit(x: Tensor, y: Tensor) {
+    fit(x: Tensor, y: Tensor): Promise<any> {
         return this.model.fit(x, y, this.fitConfig);
     }
 

tsconfig.json

@@ -6,7 +6,7 @@
     "allowUnusedLabels": false,
     "alwaysStrict": true,
     "checkJs": false,
-    "declaration": false,
+    "declaration": true,
     "diagnostics": true,
     "esModuleInterop": true,
     "forceConsistentCasingInFileNames": true,