diff --git a/requirements.txt b/requirements.txt
index 391ead3..12f3a04 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,8 +7,7 @@ ipytest
 tensorflow
 sagemaker==2.192.1
 sagemaker[local]
-black
-flake8
+ruff
 nbformat
 nbdev
 nbclient