Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions scripts/publish-npm.ts
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,11 @@ async function main() {

console.log(chalk.magenta.bold(`~~~ Preparing package ${pkg}~~~`));
console.log(chalk.magenta('~~~ Installing packages ~~~'));
// tfjs-node-gpu needs to get some files from tfjs-node.
if (pkg === "tfjs-node-gpu") {
$('yarn prep-gpu');
}

// Yarn above the other checks to make sure yarn doesn't change the lock file.
$('yarn');

Expand Down
5 changes: 4 additions & 1 deletion scripts/release-util.ts
Original file line number Diff line number Diff line change
Expand Up @@ -46,9 +46,12 @@ export const UNION_PHASE: Phase = {
deps: ['tfjs-core', 'tfjs-layers', 'tfjs-converter', 'tfjs-data']
};

// We added tfjs-core and tfjs-layers because Node has unit tests that directly
// use tf.core and tf.layers to test serialization of models. Consider moving
// the test to tf.layers.
export const NODE_PHASE: Phase = {
packages: ['tfjs-node', 'tfjs-node-gpu'],
deps: ['tfjs', 'tfjs-core'],
deps: ['tfjs', 'tfjs-core', 'tfjs-layers'],
scripts: {'tfjs-node-gpu': {'before-yarn': ['yarn prep-gpu']}}
};

Expand Down