fix: temp mpl deprec (#443)

scikit-hep · Sep 17, 2023 · cc45d29 · cc45d29
1 parent 1fcd5d9
commit cc45d29
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/src/mplhep/alice.py b/src/mplhep/alice.py
@@ -2,7 +2,7 @@
 
 import inspect
 
-from matplotlib import docstring
+from matplotlib import _docstring as docstring
 
 import mplhep
 

diff --git a/src/mplhep/atlas.py b/src/mplhep/atlas.py
@@ -15,7 +15,7 @@
 __all__ = ("style", "lumitext")
 
 
-@mpl.docstring.copy(label_base.exp_text)
+@mpl._docstring.copy(label_base.exp_text)
 def text(text="", **kwargs):
     for key, value in dict(mplhep.rcParams.text._get_kwargs()).items():
         if (
@@ -29,7 +29,7 @@ def text(text="", **kwargs):
     return label_base.exp_text("ATLAS", text=text, **kwargs)
 
 
-@mpl.docstring.copy(label_base.exp_label)
+@mpl._docstring.copy(label_base.exp_label)
 def label(label=None, **kwargs):
     for key, value in dict(mplhep.rcParams.label._get_kwargs()).items():
         if (

diff --git a/src/mplhep/cms.py b/src/mplhep/cms.py
@@ -2,7 +2,7 @@
 
 import inspect
 
-from matplotlib import docstring
+from matplotlib import _docstring as docstring
 
 import mplhep
 

diff --git a/src/mplhep/lhcb.py b/src/mplhep/lhcb.py
@@ -20,7 +20,7 @@
 
 import inspect
 
-from matplotlib import docstring
+from matplotlib import _docstring as docstring
 
 import mplhep
 from mplhep import label as label_base

diff --git a/tests/test_inputs.py b/tests/test_inputs.py
@@ -101,7 +101,7 @@ def test_inputs_bh_cat():
         bh.axis.IntCategory(range(10)), bh.axis.StrCategory("", growth=True)
     )
 
-    x = np.round(np.random.normal(5, 3, 1000))
+    x = np.round(np.random.normal(5, 3, 1000)).astype(int)
     A, Z = np.array(["A", "Z"]).view("int32")
     y = list(
         np.random.randint(low=A, high=Z, size=1000, dtype="int32").view(f"U{1000}")[0]