Add useful scripts to repo

PiperOrigin-RevId: 705308893
Change-Id: I099ef104290569385037ab0d2168b04d0f678b5e

Author: jagapiou

bin/install.sh

@@ -0,0 +1,31 @@
+#!/bin/bash
+#
+# Copyright 2024 DeepMind Technologies Limited.
+#
+# 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
+#
+#     https://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.
+#
+# Install concordia.
+set -euxo pipefail
+cd "$(dirname "$0")/.."
+
+echo 'Installing requirements...'
+pip install --no-deps --require-hashes --requirement requirements.txt
+echo
+echo
+
+echo 'Installing Concordia...'
+pip install --no-deps --no-index --no-build-isolation --editable .
+echo
+echo
+
+pip list

bin/setup_venv.sh

@@ -0,0 +1,36 @@
+#!/bin/bash
+#
+# Copyright 2024 DeepMind Technologies Limited.
+#
+# 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
+#
+#     https://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.
+#
+# Sets up the virtual environment.
+set -euo pipefail
+cd "$(dirname "$0")/.."
+
+readonly VENV_PATH="${1:-venv}"
+if [[ -d "${VENV_PATH}" ]]; then
+  read -p "Virtual environment "${VENV_PATH}" already exists. Overwrite? (Y/N) " confirm
+  [[ "${confirm}" = [Yy]* ]] && rm -rf "${VENV_PATH}" || exit 1
+fi
+
+echo "Creating virtual environment at ${VENV_PATH}..."
+python3 -m venv "${VENV_PATH}"
+source "${VENV_PATH}"/bin/activate
+python --version
+pip --version
+pip list
+echo
+echo
+
+./bin/install.sh

bin/test.sh

@@ -0,0 +1,41 @@
+#!/bin/bash
+#
+# Copyright 2024 DeepMind Technologies Limited.
+#
+# 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
+#
+#     https://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.
+#
+# Test concordia.
+set -euxo pipefail
+cd "$(dirname "$0")/.."
+FAILURES=false
+
+echo "pytest concordia..."
+pytest concordia || FAILURES=true
+echo
+echo
+
+echo "pytype concordia..."
+pytype concordia || FAILURES=true
+echo
+echo
+
+echo "pylint concordia..."
+pylint --errors-only concordia || FAILURES=true
+echo
+echo
+
+if "${FAILURES}"; then
+  echo -e '\033[0;31mFAILURE\033[0m' && exit 1
+else
+  echo -e '\033[0;32mSUCCESS\033[0m'
+fi

bin/test_examples.sh

@@ -0,0 +1,56 @@
+#!/bin/bash
+#
+# Copyright 2024 DeepMind Technologies Limited.
+#
+# 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
+#
+#     https://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.
+#
+# Test the examples.
+set -euxo pipefail
+cd "$(dirname "$0")/.."
+FAILURES=false
+
+echo "pytest examples..."
+pytest examples || FAILURES=true
+echo
+echo
+
+echo "pytype examples..."
+pytype examples || FAILURES=true
+echo
+echo
+
+echo "pylint examples..."
+pylint --errors-only examples || FAILURES=true
+echo
+echo
+
+echo "convert notebooks..."
+./bin/convert_notebooks.sh notebooks
+echo
+echo
+
+echo "pytype notebooks..."
+pytype --pythonpath=. notebooks || FAILURES=true
+echo
+echo
+
+echo "pylint notebooks..."
+pylint --errors-only notebooks || FAILURES=true
+echo
+echo
+
+if "${FAILURES}"; then
+  echo -e '\033[0;31mFAILURE\033[0m' && exit 1
+else
+  echo -e '\033[0;32mSUCCESS\033[0m'
+fi

bin/update_requirements.sh

@@ -0,0 +1,22 @@
+#!/bin/bash
+#
+# Copyright 2024 DeepMind Technologies Limited.
+#
+# 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
+#
+#     https://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.
+#
+# Update requirements.txt.
+set -euxo pipefail
+cd "$(dirname "$0")/.."
+
+pip-compile --generate-hashes --reuse-hashes --strip-extras \
+     --upgrade --extra dev setup.py examples/requirements.in