diff --git a/video_prediction_tools/utils/general_utils.py b/video_prediction_tools/utils/general_utils.py index fdb7229b33ab459cc5720b55df12ae29904914b8..1f774f54011b141eb635708486c1bf8a049b17ef 100644 --- a/video_prediction_tools/utils/general_utils.py +++ b/video_prediction_tools/utils/general_utils.py @@ -198,14 +198,14 @@ def get_era5_varatts(data_arr: xr.DataArray, name): else: addstr = "" - longname = provide_default(era5_varname_map, name_splitted[0], None) - if longname is None: + longname = provide_default(era5_varname_map, name_splitted[0], -1) + if longname == -1: pass else: data_arr["longname"] = "{0} {1}".format(longname, addstr) - unit = provide_default(era5_varunit_map, name_splitted[0], None) - if unit is None: + unit = provide_default(era5_varunit_map, name_splitted[0], -1) + if unit == -1: pass else: data_arr["unit"] = unit