From 8f911bb64e0486b733405bdba4dd7bd972ca498c Mon Sep 17 00:00:00 2001 From: Cheng Date: Thu, 25 Apr 2024 09:41:34 +0900 Subject: [PATCH] Turn off tensorflow's warning about using pure-js impl --- tests/run.ts | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/run.ts b/tests/run.ts index 476074b5..3631d353 100644 --- a/tests/run.ts +++ b/tests/run.ts @@ -2,6 +2,10 @@ import fs from 'node:fs'; import path from 'node:path'; import Mocha from 'mocha'; import yargs from 'yargs'; +import tf from '@tensorflow/tfjs'; + +// Do not warn about using pure-js tensorflow. +tf.ENV.set('IS_TEST', true); const argv = require('yargs') .string('g').alias('g', 'grep')