From 5f6d670187ebcfbca2bf73e7e28bae39c55bb47a Mon Sep 17 00:00:00 2001
From: Marcus Loren <65302107+MarcusLoppe@users.noreply.github.com>
Date: Tue, 18 Jun 2024 21:58:56 +0200
Subject: [PATCH] Update setup.py

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 7da9210..0a016ea 100644
--- a/setup.py
+++ b/setup.py
@@ -33,6 +33,7 @@
     'jaxtyping',
     'local-attention>=1.9.0',
     'numpy',
+    'matplotlib',
     'pytorch-custom-utils>=0.0.9',
     'taylor-series-linear-attention>=0.1.6',
     'torch>=2.1',