From 5518322cedad6b0f93e1021a576155084df72427 Mon Sep 17 00:00:00 2001
From: Jochem Smit <jhsmit@gmail.com>
Date: Tue, 26 Sep 2023 15:35:12 +0200
Subject: [PATCH] fix dg color scatter plotting for newer pandas versions

---
 pyhdx/plot.py | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/pyhdx/plot.py b/pyhdx/plot.py
index 97ee3693..b7932cb1 100644
--- a/pyhdx/plot.py
+++ b/pyhdx/plot.py
@@ -1,7 +1,7 @@
 from contextlib import contextmanager
 from copy import copy
 from pathlib import Path
-from typing import Type, Union
+from typing import Optional, Type, Union
 
 import matplotlib as mpl
 import matplotlib.pyplot as plt
@@ -869,16 +869,16 @@ def rainbowclouds(
 # todo this should also take Pd.Series?
 def colorbar_scatter(
     ax,
-    data,
-    y="dG",
-    yerr="covariance",
+    data: pd.DataFrame,
+    y:str="dG",
+    yerr:str="covariance",
     cmap=None,
     norm=None,
-    cbar=True,
-    cbar_kwargs=None,
-    invert_yaxis=None,
-    symmetric=False,
-    sclf=1e-3,
+    cbar:bool=True,
+    cbar_kwargs:Optional[dict]=None,
+    invert_yaxis:bool=False,
+    symmetric:bool=False,
+    sclf:float=1e-3,
     **kwargs,
 ):
     # todo make error bars optional
@@ -903,9 +903,9 @@ def colorbar_scatter(
     ax.scatter(data.index, data[y] * sclf, color=colors, **scatter_kwargs)
     with autoscale_turned_off(ax):
         ax.errorbar(
-            data.index,
-            data[y] * sclf,
-            yerr=data[yerr] * sclf,
+            np.array(data.index),
+            np.array(data[y] * sclf),
+            yerr=np.array(data[yerr] * sclf),
             zorder=-1,
             **errorbar_kwargs,
         )