Update TensorFlow.js dependency to 3.x

.env

@@ -0,0 +1,3 @@
+# Disable TensorFlow debug logging at startup.
+# Set to 0 if you want to print these debug logs.
+TF_CPP_MIN_LOG_LEVEL=3

.travis.yml

@@ -1,5 +1,5 @@
 language: node_js
-node_js: "8"
+node_js: lts/*
 script:
   - yarn lint
   - yarn test

README.md

@@ -5,11 +5,13 @@ about the file format.
 
 API:
 
-    import { parse, serialize } from "tfjs-npy"
+```ts
+import { parse, serialize } from "tfjs-npy"
 
-    parse(ab: ArrayBuffer): tf.Tensor
+function parse(ab: ArrayBuffer): tf.Tensor
 
-    serialize(tensor: tf.Tensor): Promise<ArrayBuffer>
+function serialize(tensor: tf.Tensor): Promise<ArrayBuffer>
+```
 
 https://github.com/propelml/tfjs-npy
 

npy.ts

@@ -18,9 +18,10 @@ import * as tf from "@tensorflow/tfjs-core";
 
 /** Serializes a tensor into a npy file contents. */
 export async function serialize(tensor: tf.Tensor): Promise<ArrayBuffer> {
-  const descr = new Map([["float32", "<f4"], ["int32", "<i4"]]).get(
-    tensor.dtype,
-  );
+  const descr = new Map([
+    ["float32", "<f4"],
+    ["int32", "<i4"],
+  ]).get(tensor.dtype);
 
   // First figure out how long the file is going to be so we can create the
   // output ArrayBuffer.
@@ -32,7 +33,7 @@ export async function serialize(tensor: tf.Tensor): Promise<ArrayBuffer> {
   const unpaddedLength =
     1 + magicStr.length + versionStr.length + 2 + header.length;
   // Spaces to 16-bit align.
-  const padding = " ".repeat((16 - unpaddedLength % 16) % 16);
+  const padding = " ".repeat((16 - (unpaddedLength % 16)) % 16);
   header += padding;
   assertEqual((unpaddedLength + padding.length) % 16, 0);
   // Either int32 or float32 for now Both 4 bytes per element.

npy_test.ts

@@ -12,7 +12,7 @@ limitations under the License.
 */
 
 import { test, assertEqual } from "liltest";
-import * as tf from "@tensorflow/tfjs-core";
+import * as tf from "@tensorflow/tfjs-node";
 import * as npy from "./npy";
 import { readFileSync } from "fs";
 const { expectArraysClose } = tf.test_util;

package.json

@@ -13,17 +13,20 @@
     "build": "tsc",
     "fmt": "prettier --trailing-comma all --write *.ts",
     "lint": "tslint --project tsconfig.json",
-    "test": "ts-node npy_test.ts"
+    "test": "ts-node -r dotenv/config npy_test.ts"
   },
   "dependencies": {
-    "@tensorflow/tfjs-core": "^0.9.0"
+    "@tensorflow/tfjs-core": "^3.0.0"
   },
   "devDependencies": {
-    "@types/node": "^8.10.11",
+    "@tensorflow/tfjs-node": "^3.0.0",
+    "@types/node": "^15.0.2",
+    "dotenv": "^9.0.0",
     "liltest": "^0.0.5",
-    "prettier": "^1.12.1",
-    "ts-node": "^6.0.2",
+    "prettier": "^2.2.1",
+    "seedrandom": "~2.4.3",
+    "ts-node": "^9.1.1",
     "tslint": "^5.9.1",
-    "typescript": "^2.8.3"
+    "typescript": "^4.2.4"
   }
 }

tsconfig.json

@@ -1,7 +1,7 @@
 {
   "compilerOptions": {
     "module": "commonjs",
-    "noImplicitAny": true,
+    "strict": true,
     "sourceMap": true,
     "removeComments": true,
     "preserveConstEnums": true,
@@ -11,13 +11,14 @@
     "outDir": "./dist",
     "noUnusedLocals": true,
     "noImplicitReturns": true,
-    "noImplicitThis": true,
-    "alwaysStrict": true,
     "noUnusedParameters": false,
     "pretty": true,
     "noFallthroughCasesInSwitch": true,
     "allowUnreachableCode": false,
-    "experimentalDecorators": true
+    "experimentalDecorators": true,
+    // Necessary for TensorFlow.js
+    // See https://www.tensorflow.org/js/tutorials/setup#typescript
+    "skipLibCheck": true
   },
   "include": ["*.ts"]
 }

tslint.json

@@ -38,7 +38,6 @@
     "no-require-imports": true,
     "no-string-throw": true,
     "no-unused-expression": true,
-    "no-unused-variable": true,
     "no-var-keyword": true,
     "object-literal-shorthand": true,
     "only-arrow-functions": [true, "allow-declarations", "allow-named-functions"],