tf-keras

secondmind-labs · Nov 26, 2024 · c6d4557 · c6d4557
1 parent f9f1481
commit c6d4557
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,8 @@ dependencies = [
     "notebook",
     "matplotlib",
     "taskipy",
-    "tensorflow-cpu; sys_platform == 'win32'"
+    "tensorflow-cpu; sys_platform == 'win32'",
+    "tensorflow-probability[tf]"
 ]
 
 [tool.taskipy.tasks]

diff --git a/uv.lock b/uv.lock