message: Set environment variables as part of 'build' rather than 'test' so that environments between run_ci_tests and create_wheel match
  Test Plan: Build time for create_wheel is back to 1min:
  Reviewers: #tensorflow, georgep
diff: @@ -158,10 +158,9 @@ test:rbe --test_env=USER=anon
  # Options to build TensorFlow 1.x or 2.x.
  build:v1 --define=tf_api_version=1
  build:v2 --define=tf_api_version=2
  -test:v1 --action_env=TF2_BEHAVIOR=0
  -test:v2 --action_env=TF2_BEHAVIOR=1
  +build:v1 --action_env=TF2_BEHAVIOR=0
  +build:v2 --action_env=TF2_BEHAVIOR=1
  build --config=v1
  -test --config=v1
  # Default options should come above this line
  