ranges

metno · Sep 24, 2024 · d53ff2b · d53ff2b
1 parent 8129c84
commit d53ff2b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pyaerocom/aeroval/experiment_output.py b/pyaerocom/aeroval/experiment_output.py
@@ -34,7 +34,7 @@
 from pyaerocom.stats.mda8.const import MDA8_OUTPUT_VARS
 from pyaerocom.stats.stats import _init_stats_dummy
 from pyaerocom.utils import recursive_defaultdict
-from pyaerocom.variable_helpers import get_aliases
+from pyaerocom.variable_helpers import get_aliases, get_variable
 
 MapInfo = namedtuple(
     "MapInfo",
@@ -534,7 +534,8 @@ def _get_cmap_info(self, var) -> dict[str, str | list[float]]:
             return var_ranges_defaults[var]
         try:
             varinfo = VarinfoWeb(var)
-            info = dict(scale=varinfo.cmap_bins, colmap=varinfo.cmap)
+            # TODO: get unit from pyaerocom/data/variables.ini
+            info = dict(scale=varinfo.cmap_bins, colmap=varinfo.cmap, unit=varinfo.unit)
         except (VariableDefinitionError, AttributeError):
             info = var_ranges_defaults["default"]
             logger.warning(
@@ -546,6 +547,7 @@ def _get_cmap_info(self, var) -> dict[str, str | list[float]]:
         return info
 
     def _create_var_ranges_json(self) -> None:
+        # TODO: add units
         with self.avdb.lock():
             ranges = self.avdb.get_ranges(self.proj_id, self.exp_id, default={})
 
@@ -554,6 +556,7 @@ def _create_var_ranges_json(self) -> None:
             for var in all_vars:
                 if var not in ranges or ranges[var]["scale"] == []:
                     ranges[var] = self._get_cmap_info(var)
+                ranges[var]["unit"] = get_variable(var).unit
             self.avdb.put_ranges(ranges, self.proj_id, self.exp_id)
 
     def _create_statistics_json(self) -> None: