diff --git a/pyproject.toml b/pyproject.toml
index 7cd06d786..ab7603535 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,16 +11,11 @@ dependencies = [
   "pymysql>=0.7.2",
   "deepdiff",
   "pyparsing",
-  "ipython",
   "pandas",
   "tqdm",
   "networkx",
   "pydot",
   "fsspec>=2023.1.0",
-  "matplotlib",
-  "faker",
-  "urllib3",
-  "setuptools",
   "pydantic-settings>=2.0.0",
 ]
 
@@ -88,6 +83,9 @@ test = [
   "pytest",
   "pytest-cov",
   "requests",
+  "faker",
+  "matplotlib",
+  "ipython",
   "graphviz",
   "testcontainers[mysql,minio]>=4.0",
   "polars>=0.20.0",
@@ -100,10 +98,14 @@ gcs = ["gcsfs>=2023.1.0"]
 azure = ["adlfs>=2023.1.0"]
 polars = ["polars>=0.20.0"]
 arrow = ["pyarrow>=14.0.0"]
+viz = ["matplotlib", "ipython"]
 test = [
   "pytest",
   "pytest-cov",
   "requests",
+  "faker",
+  "matplotlib",
+  "ipython",
   "s3fs>=2023.1.0",
   "testcontainers[mysql,minio]>=4.0",
   "polars>=0.20.0",