From da3f049d9d6cbfa7f58d9c7b680d5c57ff2250a8 Mon Sep 17 00:00:00 2001
From: elijahbenizzy <elijah_ben_izzy@alumni.brown.edu>
Date: Wed, 25 Oct 2023 14:04:01 -0700
Subject: [PATCH] Moves matplotlib outside of requirements.txt

---
 hamilton/plugins/sklearn_plot_extensions.py | 1 +
 requirements-test.txt                       | 1 +
 requirements.txt                            | 1 -
 3 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/hamilton/plugins/sklearn_plot_extensions.py b/hamilton/plugins/sklearn_plot_extensions.py
index 4380ad329..2f7cec609 100644
--- a/hamilton/plugins/sklearn_plot_extensions.py
+++ b/hamilton/plugins/sklearn_plot_extensions.py
@@ -3,6 +3,7 @@
 from typing import Any, Collection, Dict, Optional, Type, Union
 
 try:
+    import matplotlib  # noqa: F401
     import sklearn.inspection
     import sklearn.metrics
     import sklearn.model_selection
diff --git a/requirements-test.txt b/requirements-test.txt
index f4d02951a..dc8a2d368 100644
--- a/requirements-test.txt
+++ b/requirements-test.txt
@@ -2,6 +2,7 @@ graphviz
 lightgbm
 lxml
 lz4
+matplotlib
 networkx
 polars
 pyarrow
diff --git a/requirements.txt b/requirements.txt
index b3e3cd7c7..dd730b394 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,3 @@
-matplotlib
 numpy
 pandas
 typing_extensions > 4.0.0