chore: adding the installation helper in lib/index.ts

machinelearnjs · Dec 28, 2018 · 1061bdf · 1061bdf
1 parent ed03a14
commit 1061bdf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/lib/index.ts b/src/lib/index.ts
@@ -1,3 +1,6 @@
+import { installTF } from './utils/deps';
+installTF();
+
 import * as cluster from './cluster';
 import * as datasets from './datasets';
 import * as decomposition from './decomposition';
@@ -11,9 +14,6 @@ import * as preprocessing from './preprocessing';
 import * as svm from './svm';
 import * as tree from './tree';
 
-// Calling tfjs-node
-import '@tensorflow/tfjs-node';
-
 // prettier-ignore
 export {
 	cluster,