diff --git a/setup.py b/setup.py
index a2a4992..16abb58 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@ def readme():
         "tensorflow": ["tensorflow>=2.4.0"],
         "tensorflow_gpu": ["tensorflow-gpu>=2.4.0"],
         "test": [
-            "black==23.12.1",
+            "black==24.4.0",
             "dill==0.3.8",
             "flake8==7.0.0",
             "isort==5.12.0",