Add TF extra for TFP

wesselb · Apr 20, 2024 · 56b673b · 56b673b
1 parent 3f064bb
commit 56b673b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,7 +40,7 @@ dev = [
     "pyright",
     "torch",
     "tensorflow",
-    "tensorflow-probability",
+    "tensorflow-probability[tf]",
 ]
 
 [project.urls]