fix: @tensorflow/tfjs-node

.github/workflows/ci.yml

@@ -17,8 +17,8 @@ jobs:
           # Active LTS + other OS
           - os: macos-latest
             node_version: 18
-          # - os: windows-latest  @tensorflow/tfjs-node in window has problem
-          #   node_version: 18
+          - os: windows-latest
+            node_version: 18
       fail-fast: false
     name: "Build&Test: node-${{ matrix.node_version }}, ${{ matrix.os }}"
     steps:
@@ -39,6 +39,9 @@ jobs:
         run: |
           echo "STORE_PATH=$(pnpm store path)" >> $GITHUB_OUTPUT
 
+      - name: prepare
+        run: pnpm pre
+
       - name: Install deps
         run: pnpm install
 

.prettierignore

@@ -2,4 +2,5 @@ dist/
 LICENSE.md
 pnpm-lock.yaml
 pnpm-workspace.yaml
-assets
+assets
+.vscode

bin/preinstall.sh

@@ -0,0 +1,2 @@
+# @tensorflow/tfjs-node in github actions
+pnpm i -g node-gyp

package.json

@@ -7,6 +7,7 @@
     "node": ">=16.0.0"
   },
   "scripts": {
+    "pre": "sh ./bin/preinstall.sh",
     "preinstall": "npx only-allow pnpm",
     "clean": "sh ./bin/clean.sh",
     "build": "sh ./bin/build.sh",

packages/ml/client/index.tsx

@@ -2,6 +2,7 @@ import React from 'react';
 import { createRoot } from 'react-dom/client';
 import { BrowserRouter } from 'react-router-dom';
 import * as tf from '@tensorflow/tfjs';
+import { createLotsOfTensors } from './lib';
 
 const App = () => {
   // Define a model for linear regression.

packages/ml/client/lib.ts

@@ -6,4 +6,6 @@ export function createLotsOfTensors(): void {
         const b = tf.scalar(i)
         a.mul(b).print()
     }
-}
+}
+
+export const cleanMemoryCreateLotsOfTensors = tf.tidy(createLotsOfTensors)