Remove datetime workaround logic

aospy/calc.py

@@ -82,7 +82,9 @@ def _file_name(self, dtype_out_time, extension='nc'):
                                           dtype_vert=self.dtype_out_vert)
         in_lbl = utils.io.data_in_label(self.intvl_in, self.dtype_in_time,
                                         self.dtype_in_vert)
-        yr_lbl = utils.io.yr_label((self.start_date.year, self.end_date.year))
+        start_year = utils.times.infer_year(self.start_date)
+        end_year = utils.times.infer_year(self.end_date)
+        yr_lbl = utils.io.yr_label((start_year, end_year))
         return '.'.join(
             [self.name, out_lbl, in_lbl, self.model.name,
              self.run.name, yr_lbl, extension]

aospy/data_loader.py

@@ -1,7 +1,6 @@
 """aospy DataLoader objects"""
 import logging
 import os
-import warnings
 
 import numpy as np
 import xarray as xr
@@ -180,7 +179,6 @@ def _prep_time_data(ds):
         The processed Dataset and minimum and maximum years in the loaded data
     """
     ds = times.ensure_time_as_index(ds)
-    ds, min_year, max_year = times.numpy_datetime_workaround_encode_cf(ds)
     if TIME_BOUNDS_STR in ds:
         ds = times.ensure_time_avg_has_cf_metadata(ds)
         ds[TIME_STR] = times.average_time_bounds(ds)
@@ -189,10 +187,10 @@ def _prep_time_data(ds):
                         "values in time, even though this may not be "
                         "the case")
         ds = times.add_uniform_time_weights(ds)
-    with warnings.catch_warnings(record=True):
+    with xr.set_options(enable_cftimeindex=True):
         ds = xr.decode_cf(ds, decode_times=True, decode_coords=False,
                           mask_and_scale=True)
-    return ds, min_year, max_year
+    return ds
 
 
 def _load_data_from_disk(file_set, preprocess_func=lambda ds: ds,
@@ -280,17 +278,15 @@ def load_variable(self, var=None, start_date=None, end_date=None,
             coords=self.coords, start_date=start_date, end_date=end_date,
             time_offset=time_offset, **DataAttrs
         )
-
-        ds, min_year, max_year = _prep_time_data(ds)
+        ds = _prep_time_data(ds)
+        start_date = times.maybe_convert_to_index_date_type(
+            ds.indexes[TIME_STR], start_date)
+        end_date = times.maybe_convert_to_index_date_type(
+            ds.indexes[TIME_STR], end_date)
         ds = set_grid_attrs_as_coords(ds)
         da = _sel_var(ds, var, self.upcast_float32)
         da = self._maybe_apply_time_shift(da, time_offset, **DataAttrs)
-
-        start_date_xarray = times.numpy_datetime_range_workaround(
-            start_date, min_year, max_year)
-        end_date_xarray = start_date_xarray + (end_date - start_date)
-        return times.sel_time(da, np.datetime64(start_date_xarray),
-                              np.datetime64(end_date_xarray)).load()
+        return times.sel_time(da, start_date, end_date).load()
 
     def recursively_compute_variable(self, var, start_date=None, end_date=None,
                                      time_offset=None, **DataAttrs):
@@ -609,10 +605,13 @@ def _input_data_paths_gfdl(self, name, start_date, end_date, domain,
         else:
             subdir = os.path.join(intvl_in, dur_str)
         direc = os.path.join(self.data_direc, domain, dtype_lbl, subdir)
+        data_start_year = times.infer_year(self.data_start_date)
+        start_year = times.infer_year(start_date)
+        end_year = times.infer_year(end_date)
         files = [os.path.join(direc, io.data_name_gfdl(
                     name, domain, dtype, intvl_in, year, intvl_out,
-                    self.data_start_date.year, self.data_dur))
-                 for year in range(start_date.year, end_date.year + 1)]
+                    data_start_year, self.data_dur))
+                 for year in range(start_year, end_year + 1)]
         files = list(set(files))
         files.sort()
         return files

aospy/examples/example_obj_lib.py

@@ -1,5 +1,5 @@
 """Sample aospy object library using the included example data."""
-import datetime
+from datetime import datetime
 import os
 
 import aospy
@@ -17,8 +17,8 @@
     description=(
         'Control simulation of the idealized moist model'
     ),
-    default_start_date=datetime.datetime(4, 1, 1),
-    default_end_date=datetime.datetime(6, 12, 31),
+    default_start_date=datetime(4, 1, 1),
+    default_end_date=datetime(6, 12, 31),
     data_loader=DictDataLoader(_file_map)
 )
 

aospy/test/data/objects/examples.py

@@ -1,6 +1,7 @@
-from datetime import datetime
 import os
 
+from cftime import DatetimeNoLeap
+
 from aospy import Proj, Model, Run, Var, Region
 from aospy.data_loader import NestedDictDataLoader
 
@@ -25,8 +26,8 @@ def total_precipitation(convection_rain, condensation_rain):
         'Control simulation of the idealized moist model'
     ),
     data_loader=NestedDictDataLoader(file_map),
-    default_start_date=datetime(4, 1, 1),
-    default_end_date=datetime(6, 12, 31)
+    default_start_date=DatetimeNoLeap(4, 1, 1),
+    default_end_date=DatetimeNoLeap(6, 12, 31)
 )
 
 example_model = Model(