diff --git a/spinoffs/autobnn/README.md b/spinoffs/autobnn/README.md index 843259793d..53b3f7f2ed 100644 --- a/spinoffs/autobnn/README.md +++ b/spinoffs/autobnn/README.md @@ -18,8 +18,13 @@ Vikash K. Mansinghka. _Sequential Monte Carlo Learning for Time Series Structure Discovery_. ICML 2023. -## Setup +## Installation -AutoBNN has three additional dependencies beyond those used by the core -Tensorflow Probability package: flax, scipy and jaxtyping. These -can be installed by running `setup_autobnn.sh`. +AutoBNN can be installed with pip + +``` +pip install autobnn +``` + +or it can be installed by source by following [these instructions] +(https://github.com/tensorflow/probability?tab=readme-ov-file#installing-from-source). diff --git a/spinoffs/autobnn/setup_autobnn.sh b/spinoffs/autobnn/setup_autobnn.sh deleted file mode 100755 index fdb4ae7e33..0000000000 --- a/spinoffs/autobnn/setup_autobnn.sh +++ /dev/null @@ -1,32 +0,0 @@ -#!/usr/bin/env bash -# Copyright 2023 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. -# ============================================================================ - -set -v # print commands as they are executed -set -e # fail and exit on any command erroring - -user_flag_is_set() { - # We could use getopts but it's annoying and arguably overkill here. We should - # consider it if ever this script grows more complicated. - [[ "$SCRIPT_ARGS" =~ --user ]] -} - -if user_flag_set; then - PIP_FLAGS="--user" -else - PIP_FLAGS="" -fi - -python -m pip install $PIP_FLAGS bayeux-ml chex flax jaxtyping matplotlib pandas scipy