diff --git a/prepper/exportable.py b/prepper/exportable.py index ce1bab3..ab63042 100644 --- a/prepper/exportable.py +++ b/prepper/exportable.py @@ -25,11 +25,6 @@ from prepper.enums import H5StoreTypes from prepper.utils import check_equality -try: - from auto_uncertainties import NumpyDowncastWarning -except ImportError: - NumpyDowncastWarning = None - __all__ = [ "ExportableClassMixin", ] @@ -41,7 +36,7 @@ class ExportableClassMixin(metaclass=ABCMeta): """ - _constructor_args: Dict[str, Any] = {} + _constructor_args: Dict[str, Any] api_version: float _exportable_attributes: List[str] _exportable_functions: List[str] @@ -235,10 +230,6 @@ def to_hdf5(self, path): warnings.simplefilter( "ignore", category=np.VisibleDeprecationWarning ) - if NumpyDowncastWarning is not None: - warnings.simplefilter( - "ignore", category=NumpyDowncastWarning - ) self._write_hdf5_contents( temp_file, group="/", existing_groups={} ) diff --git a/prepper/io_handlers.py b/prepper/io_handlers.py index b3c2532..b6d1213 100644 --- a/prepper/io_handlers.py +++ b/prepper/io_handlers.py @@ -842,7 +842,7 @@ def load_unit_or_error_ndarrays(file: str, group: str): raise H5StoreException try: v = v[0] - except IndexError: + except (IndexError, TypeError): pass return v