Turn off tensorflow's warning about using pure-js impl

frost-beta · Apr 25, 2024 · 8f911bb · 8f911bb
1 parent 7677dd4
commit 8f911bb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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')