Add plotly and tensorflow_probability
This lets us add plots to aim, and also adds better probability
distribution support.
Change-Id: I6a01afa5fc831bbaa1e9af9c62af5086616056da
Signed-off-by: Austin Schuh <austin.linux@gmail.com>
diff --git a/tools/python/requirements.txt b/tools/python/requirements.txt
index 241425a..342d8ab 100644
--- a/tools/python/requirements.txt
+++ b/tools/python/requirements.txt
@@ -41,9 +41,12 @@
tensorflow
tensorflow_datasets
+tensorflow_probability
+tf_keras
# Experience buffer for reinforcement learning
flashbax
# Experiment tracking
aim
+plotly