diff --git a/pkgs/development/python-modules/stable-baselines3/default.nix b/pkgs/development/python-modules/stable-baselines3/default.nix
index d4e72d1ee173..b75d6a126581 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
   ];
 
@@ -83,6 +85,10 @@ buildPythonPackage (finalAttrs: {
     # RuntimeError: *** -[__NSPlaceholderArray initWithObjects:count:]: attempt to insert nil object from objects[1]
     "test_report_figure_to_tensorboard"
     "test_unsupported_figure_format"
+
+    # Doesn't manage to find tqdm and rich from nativeCheckInputs
+    # ImportError: You must install tqdm and rich in order to use the progress bar callback
+    "test_callbacks"
   ];
 
   meta = {