ENH: Guard against tz-naive index for Panel data.

2026-06-30 17:05:35 +08:00 · 2016-07-11 16:20:34 -04:00
parent bdce4ef257
commit 69506570dd
1 changed files with 4 additions and 0 deletions
@@ -612,6 +612,10 @@ class TradingAlgorithm(object):
                data = data.swapaxes(0, 2)

            if isinstance(data, pd.Panel):
+                # Guard against tz-naive index.
+                if data.major_axis.tz is None:
+                    data.major_axis = data.major_axis.tz_localize('UTC')
+
                # For compatibility with existing examples allow start/end
                # to be inferred.
                if overwrite_sim_params: