update variables

ecmwf · Oct 25, 2024 · 8d9e071 · 8d9e071
1 parent 1cd2642
commit 8d9e071
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/anemoi/transform/variables/__init__.py b/src/anemoi/transform/variables/__init__.py
@@ -54,3 +54,7 @@ def level(self):
     @abstractmethod
     def is_constant_in_time(self):
         pass
+
+    @property
+    def grib_keys(self):
+        raise NotImplementedError(f"Method `grib_keys` not implemented for {self.__class__.__name__}")
diff --git a/src/anemoi/transform/variables/variables.py b/src/anemoi/transform/variables/variables.py
@@ -37,6 +37,14 @@ def is_from_input(self):
     def is_computed_forcing(self):
         return self.data.get("computed_forcing", False)
 
+    @property
+    def is_accumulation(self):
+        return self.data.get("process") == "accumulation"
+
+    @property
+    def grib_keys(self):
+        return self.data.get("mars", {})
+
 
 class VariableFromDict(VariableFromMarsVocabulary):
     """A variable that is defined by a user provided dictionary."""