From 02202f8dc40e65f4bb45fe9661114c5fb3a81077 Mon Sep 17 00:00:00 2001
From: Wessel Bruinsma <wessel.p.bruinsma@gmail.com>
Date: Sun, 15 Dec 2024 15:21:01 +0100
Subject: [PATCH] Fix TF2 dep

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 0d13788..5da6ba0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ pytest-cov
 coveralls
 autograd
 tensorflow>=2
-tensorflow-probability
+tensorflow-probability[tf]
 torch
 jax
 jaxlib