diff --git a/setup.cfg b/setup.cfg
index d825466..80f15c6 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -22,6 +22,7 @@ install_requires=
     torch>=1.12.1
     PyYAML>=6.0
     tqdm
+    matplotlib
 
 [options.package_data]
 * = data/*