Fix "Legend.draggable() is deprecated" warning

mahdinobar · Nov 24, 2018 · e49471a · e49471a
1 parent dd0f628
commit e49471a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spinup/utils/plot.py b/spinup/utils/plot.py
@@ -39,7 +39,7 @@ def plot_data(data, xaxis='Epoch', value="AverageEpRet", condition="Condition1",
 
     Changes the colorscheme and the default legend style, though.
     """
-    plt.legend(loc='best').draggable()
+    plt.legend(loc='best').set_draggable(True)
 
     """
     For the version of the legend used in the Spinning Up benchmarking page,