Update e2e dependencies in a release PR (#6254)

Update e2e dependencies to the versions that are released when releasing the tfjs monorepo. This fixes a failure in release tests where e2e tries to install 'link:' dependencies.

Add a '--local' option to the release-tfjs script to prevent creating a release PR. This is just for development and makes it easier to see the output of the script.

Author: mattsoulanille

scripts/release-tfjs.ts

@@ -27,7 +27,7 @@ import * as argparse from 'argparse';
 import chalk from 'chalk';
 import * as fs from 'fs';
 import * as shell from 'shelljs';
-import {TMP_DIR, $, question, makeReleaseDir, createPR, TFJS_RELEASE_UNIT, updateTFJSDependencyVersions, ALPHA_RELEASE_UNIT, getMinorUpdateVersion, getPatchUpdateVersion} from './release-util';
+import {TMP_DIR, $, question, makeReleaseDir, createPR, TFJS_RELEASE_UNIT, updateTFJSDependencyVersions, ALPHA_RELEASE_UNIT, getMinorUpdateVersion, getPatchUpdateVersion, E2E_PHASE} from './release-util';
 
 const parser = new argparse.ArgumentParser();
 
@@ -36,6 +36,11 @@ parser.addArgument('--git-protocol', {
   help: 'Use the git protocol rather than the http protocol when cloning repos.'
 });
 
+parser.addArgument('--local', {
+  action: 'storeTrue',
+  help: 'Only create the release branch locally. Do not push or create a PR.',
+});
+
 async function main() {
   const args = parser.parseArgs();
   const urlBase = args.git_protocol ? 'git@github.com:' : 'https://github.com/';
@@ -86,10 +91,14 @@ async function main() {
   console.log(chalk.magenta.bold(
       `~~~ Creating new release branch ${releaseBranch} ~~~`));
   $(`git checkout -b ${releaseBranch} ${commit}`);
-  $(`git push origin ${releaseBranch}`);
+  if (!args.local) {
+    $(`git push origin ${releaseBranch}`);
+  }
 
   // Update versions in package.json files.
-  const phases = [...TFJS_RELEASE_UNIT.phases, ...ALPHA_RELEASE_UNIT.phases];
+  const phases = [
+    ...TFJS_RELEASE_UNIT.phases, ...ALPHA_RELEASE_UNIT.phases, E2E_PHASE
+  ];
   for (const phase of phases) {
     for (const packageName of phase.packages) {
       shell.cd(packageName);
@@ -109,8 +118,8 @@ async function main() {
 
       shell.cd('..');
 
-      // Make version for all packages other than tfjs-node-gpu.
-      if (packageName !== 'tfjs-node-gpu') {
+      // Make version for all packages other than tfjs-node-gpu and e2e.
+      if (packageName !== 'tfjs-node-gpu' && packageName !== 'e2e') {
         $(`./scripts/make-version.js ${packageName}`);
       }
     }
@@ -120,7 +129,9 @@ async function main() {
   const devBranchName = `dev_${releaseBranch}`;
 
   const message = `Update monorepo to ${newVersion}.`;
-  createPR(devBranchName, releaseBranch, message);
+  if (!args.local) {
+    createPR(devBranchName, releaseBranch, message);
+  }
 
   console.log(
       'Done. FYI, this script does not publish to NPM. ' +
@@ -131,6 +142,9 @@ async function main() {
       'Please remeber to update the website once you have released ' +
       'a new package version.');
 
+  if (args.local) {
+    console.log(`Local output located in ${dir}`)
+  }
   process.exit(0);
 }
 

scripts/release-util.ts

@@ -130,6 +130,16 @@ export const WEBSITE_PHASE: Phase = {
   title: 'Update website to latest dependencies.'
 };
 
+// Note that e2e is not actually published. As a result, this phase is not
+// included in any release unit, however, it is used for updating dependencies.
+export const E2E_PHASE: Phase = {
+  packages: ['e2e'],
+  deps: [
+    'tfjs', 'tfjs-backend-cpu', 'tfjs-backend-wasm', 'tfjs-backend-webgl',
+    'tfjs-converter', 'tfjs-core', 'tfjs-data', 'tfjs-layers', 'tfjs-node'
+  ],
+}
+
 export const TFJS_RELEASE_UNIT: ReleaseUnit = {
   name: 'tfjs',
   phases: [