Fix Forecast plot method (awslabs#3006)

Latest matplotlib update had broken it since we were using internal API
kashif · Sep 15, 2023 · c8d499e · c8d499e
1 parent bb7c7c1
commit c8d499e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/requirements/requirements-docs.txt b/requirements/requirements-docs.txt
@@ -15,4 +15,4 @@ click
 orjson
 black
 holidays~=0.9
-matplotlib
+matplotlib~=3.6
diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
@@ -8,4 +8,4 @@ ujson
 orjson
 requests
 holidays~=0.9
-matplotlib
+matplotlib~=3.6
diff --git a/src/gluonts/model/forecast.py b/src/gluonts/model/forecast.py
@@ -316,7 +316,7 @@ def plot(
         # If no color is provided, we use matplotlib's internal color cycle.
         # Note: This is an internal API and might change in the future.
         color = maybe.unwrap_or_else(
-            color, lambda: next(ax._get_lines.prop_cycler)["color"]
+            color, lambda: ax._get_lines.get_next_color()
         )
 
         # Plot median forecast