bugfixes

varchasgopalaswamy · Apr 23, 2024 · 65d65b9 · 65d65b9
1 parent 7d01efa
commit 65d65b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 11 deletions.
diff --git 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
@@ -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