Revert to datetime64[ms]

equinor · Nov 15, 2023 · ac506e7 · ac506e7
1 parent 663daa0
commit ac506e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ecl2df/wellcompletiondata.py b/ecl2df/wellcompletiondata.py
@@ -228,7 +228,7 @@ def _df2pyarrow(dframe: pd.DataFrame) -> pyarrow.Table:
         else:
             field_metadata = {}
         if colname == "DATE":
-            dtype = pyarrow.timestamp("ns")
+            dtype = pyarrow.timestamp("ms")
         elif pd.api.types.is_integer_dtype(dframe.dtypes[colname]):
             dtype = pyarrow.int32()
         elif pd.api.types.is_string_dtype(dframe.dtypes[colname]):

diff --git a/tests/test_wellcompletiondata.py b/tests/test_wellcompletiondata.py
@@ -86,7 +86,7 @@ def test_df2pyarrow():
         eclfiles, zonemap=EIGHTCELLS_ZONEMAP, use_wellconnstatus=False
     )
     df["KH"] = df["KH"].astype(np.int32)
-    pd.testing.assert_frame_equal(df, _df2pyarrow(df).to_pandas(), check_like=True)
+    pd.testing.assert_frame_equal(df, _df2pyarrow(df).to_pandas(), check_like=True, check_dtype=False)
 
 
 def test_metadata():