group paths in a constants.js file

Author: lucleray

constants.js

@@ -0,0 +1,6 @@
+const { resolve: pathResolve } = require('path')
+
+const TFJS_PATH = '/tmp/tfjs-node'
+const TAR_PATH = pathResolve(__dirname, 'tfjs-node.br')
+
+module.exports = { TFJS_PATH, TAR_PATH }

index.js

@@ -1,10 +1,7 @@
 const tar = require('tar')
 const iltorb = require('iltorb')
 const fs = require('fs')
-const { resolve: pathResolve } = require('path')
-
-const TFJS_PATH = '/tmp/tfjs-node'
-const TAR_PATH = pathResolve(__dirname, 'tfjs-node.br')
+const { TFJS_PATH, TAR_PATH } = require('./constants')
 
 // this hack is required to avoid webpack/rollup/... bundling the required path
 const requireFunc =

package.json

@@ -1,6 +1,6 @@
 {
   "name": "tfjs-lambda",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "main": "index.js",
   "license": "MIT",
   "files": [

readme.md

@@ -18,6 +18,6 @@ During cold start, the files are deflated in `/tmp` and required in your node pr
 
 ## Motivation
 
-`@tensorflow/tfjs` works with AWS Lambda but the main problem is that it is slow very slow when used in node. On the other hand, `@tensorflow/tfjs-node` is fast when used with node but it is >140mo and it does not fit under AWS Lambda's size limit (50mo).
+`@tensorflow/tfjs` works with AWS Lambda but the main problem is that it is slow very slow when used in node. On the other hand, `@tensorflow/tfjs-node` is fast when used with node but it is >140mo and it does not fit under AWS Lambda's size limit (50mo) and it needs to be pre-compiled for lambda for it to work in a lambda environment.
 
 I was looking for an easy way to use tensorflowjs with lambda and I couldn't find any, so I made this package.

scripts/compress.js

@@ -1,8 +1,9 @@
 const tar = require('tar')
 const iltorb = require('iltorb')
 const fs = require('fs')
+const { TFJS_PATH, TAR_PATH } = require('../constants')
 
 tar
-  .c({ cwd: '/tmp/tfjs-node' }, ['index.js', 'node_modules'])
+  .c({ cwd: TFJS_PATH }, ['index.js', 'node_modules'])
   .pipe(iltorb.compressStream())
-  .pipe(fs.createWriteStream('../tfjs-node.br'))
+  .pipe(fs.createWriteStream(TAR_PATH))