diff --git a/pkgs/development/python-modules/stable-baselines3/default.nix b/pkgs/development/python-modules/stable-baselines3/default.nix
index d4e72d1ee173..7e7a1f927219 100644
--- a/pkgs/development/python-modules/stable-baselines3/default.nix
+++ b/pkgs/development/python-modules/stable-baselines3/default.nix
@@ -19,6 +19,7 @@
   ale-py,
   pytestCheckHook,
   rich,
+  tensorboard,
   tqdm,
 }:
 buildPythonPackage (finalAttrs: {
@@ -52,6 +53,7 @@ buildPythonPackage (finalAttrs: {
     ale-py
     pytestCheckHook
     rich
+    tensorboard
     tqdm
   ];