Added missing filename key.

See #1066
Open-MSS · Jun 30, 2021 · 0203aeb · 0203aeb
1 parent b707b2f
commit 0203aeb
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/mslib/mswms/dataaccess.py b/mslib/mswms/dataaccess.py
@@ -225,7 +225,7 @@ def _determine_filename(self, variable, vartype, init_time, valid_time, reload=T
                 raise ValueError(f"variable type {vartype} not available for variable {variable}")
 
     def _parse_file(self, filename):
-        elevations = {"levels": [], "units": None}
+        elevations = {"filename": filename, "levels": [], "units": None}
         with netCDF4.Dataset(os.path.join(self._root_path, filename)) as dataset:
             time_name, time_var = netCDF4tools.identify_CF_time(dataset)
             init_time = netCDF4tools.num2date(0, time_var.units)
@@ -248,7 +248,10 @@ def _parse_file(self, filename):
                 raise IOError("Problem with longitude coordinate variable")
 
             if vert_type != "sfc":
-                elevations = {"levels": vert_var[:], "units": getattr(vert_var, "units", "1")}
+                elevations = {
+                    "filename": filename,
+                    "levels": vert_var[:],
+                    "units": getattr(vert_var, "units", "dimensionless")}
                 if vert_type in self._elevations:
                     if len(vert_var[:]) != len(self._elevations[vert_type]["levels"]):
                         raise IOError(f"Number of vertical levels does not fit to levels of "
@@ -419,7 +422,7 @@ def setup(self):
                 del self._file_cache[filename]
 
         self._filetree = {}
-        self._elevations = {"sfc": {"filename": None, "levels": []}}
+        self._elevations = {"sfc": {"filename": None, "levels": [], "units": None}}
 
         # Build the tree structure.
         for filename in self._available_files: