opf: add a work around for pandas bug #12050

openego · Jan 18, 2016 · dac9299 · dac9299
1 parent 12d8120
commit dac9299
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/opf.py b/opf.py
@@ -38,6 +38,7 @@
 from .pf import calculate_dependent_values, find_slack_bus
 
 from itertools import chain
+from distutils.version import StrictVersion
 
 import pandas as pd
 
@@ -410,6 +411,9 @@ def define_linear_objective(network,snapshots):
 
 
 def extract_optimisation_results(network,snapshots):
+    if isinstance(snapshots, pd.DatetimeIndex) and StrictVersion(pd.version.version) < '0.18.0':
+        # Work around pandas bug #12050 (https://github.com/pydata/pandas/issues/12050)
+        snapshots = list(snapshots)
 
     #get value of objective function
     network.objective = network.results["Problem"][0]["Lower bound"]