Merge pull request #10 from UnravelSports/feat/pressing-intensity

fix mpl

examples/pressing_intensity.ipynb

@@ -89,6 +89,13 @@
     "# %pip install unravelsports --quiet"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "You might need to `pip install ffmpeg mplsoccer matplotlib seaborn` for this to work."
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -508,9 +515,7 @@
    "source": [
     "**Video**\n",
     "\n",
-    "Finally, we render our 35 second video using `FuncAnimation` and `mplsoccer`.\n",
-    "\n",
-    "You might need to `pip install ffmpeg` for this to work."
+    "Finally, we render our 35 second video using `FuncAnimation` and `mplsoccer`."
    ]
   },
   {

unravel/soccer/models/pressing_intensity.py

@@ -13,10 +13,6 @@
     Constant,
 )
 
-from mplsoccer import VerticalPitch, Pitch
-from matplotlib.axes import Axes
-from matplotlib.figure import Figure
-
 from .utils import time_to_intercept, probability_to_intercept
 
 

unravel/utils/display/colors.py

@@ -3,8 +3,6 @@
 
 import re
 
-from matplotlib.colors import LinearSegmentedColormap
-
 
 @dataclass
 class Color:
@@ -72,5 +70,26 @@ class GameColors:
 
 @dataclass
 class ColorMaps:
-    YELLOW_RED = LinearSegmentedColormap.from_list("", YlRd)
-    YELLOW_RED_R = LinearSegmentedColormap.from_list("", list(reversed(YlRd)))
+    _YlRd = ["#FFFF00", "#FF0000"]  # Replace with actual YlRd values
+
+    @property
+    def YELLOW_RED(self):
+        try:
+            from matplotlib.colors import LinearSegmentedColormap
+
+            return LinearSegmentedColormap.from_list("", self._YlRd)
+        except ImportError:
+            raise ImportError(
+                "Seems like you don't have matplotlib installed. Please install it using: pip install matplotlib"
+            )
+
+    @property
+    def YELLOW_RED_R(self):
+        try:
+            from matplotlib.colors import LinearSegmentedColormap
+
+            return LinearSegmentedColormap.from_list("", list(reversed(self._YlRd)))
+        except ImportError:
+            raise ImportError(
+                "Seems like you don't have matplotlib installed. Please install it using: pip install matplotlib"
+            )