Circumvent Pandas bugs for years > 2262

equinor · Jan 19, 2023 · 5af2dc6 · 5af2dc6
1 parent 338df6d
commit 5af2dc6
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/python/ecl/summary/ecl_sum.py b/python/ecl/summary/ecl_sum.py
@@ -581,7 +581,7 @@ def pandas_frame(self, time_index=None, column_keys=None):
             raise ValueError("No valid key")
 
         if time_index is None:
-            time_index = self.numpy_dates
+            time_index = self.dates
             data = numpy.zeros([len(time_index), len(keywords)])
             EclSum._init_pandas_frame(
                 self, keywords, data.ctypes.data_as(ctypes.POINTER(ctypes.c_double))
@@ -635,16 +635,21 @@ def _compile_headers_list(headers, dims):
     @classmethod
     def from_pandas(cls, case, frame, dims=None, headers=None):
         start_time = frame.index[0]
+
+        # Avoid Pandas or numpy timestamps, to avoid Pandas attempting to create
+        # timestamp64[ns] indices (which can't go beyond year 2262)
+        # https://github.com/pandas-dev/pandas/issues/39727
+        if isinstance(start_time, pandas.Timestamp):
+            start_time = start_time.to_pydatetime()
+
         var_list = []
         if headers is None:
             header_list = EclSum._compile_headers_list(frame.columns.values, dims)
         else:
             header_list = EclSum._compile_headers_list(headers, dims)
         if dims is None:
             dims = [1, 1, 1]
-        ecl_sum = EclSum.writer(
-            case, start_time.to_pydatetime(), dims[0], dims[1], dims[2]
-        )
+        ecl_sum = EclSum.writer(case, start_time, dims[0], dims[1], dims[2])
         for kw, wgname, num, unit in header_list:
             var_list.append(
                 ecl_sum.addVariable(kw, wgname=wgname, num=num, unit=unit).getKey1()