Tests

In open source tests, use tf-keras-nightly (Keras 2) instead of Keras 3. #2034

	# Copyright 2020 The TensorFlow Probability Authors.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.
	# ============================================================================
	name: Tests
	on: [push, pull_request]
	jobs:
	lints:
	name: Lints
	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: [3.9]
	steps:
	- name: Checkout
	uses: actions/checkout@v1
	with:
	fetch-depth: 20
	- name: Setup Python
	uses: actions/setup-python@v2
	with:
	python-version: ${{ matrix.python-version }}
	- name: Lints
	run: \|
	./testing/run_github_lints.sh
	tests:
	name: Tests
	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: [3.9]
	shard: [0, 1, 2, 3, 4]
	env:
	SHARD: ${{ matrix.shard }}
	NUM_SHARDS: 5
	steps:
	- name: Checkout
	uses: actions/checkout@v1
	with:
	fetch-depth: 1
	- name: Setup Python
	uses: actions/setup-python@v2
	with:
	python-version: ${{ matrix.python-version }}
	- name: Tests
	run: \|
	./testing/run_github_tests.sh
	- name: Upload test logs
	if: failure()
	uses: actions/upload-artifact@v1
	with:
	name: testlogs-${{ matrix.shard }}
	path: bazel-testlogs

Provide feedback