[postprocessor] Added oct_vect_getter in generic rules

2021-04-20 10:56:38 +02:00
parent d57ef50402
commit 496261068b
2 changed files with 13 additions and 1 deletions
@@ -32,7 +32,14 @@ import pspec_new


 from units import U
-from baseprocessor import HDF5Container, Rule, norm_getter, simple_getter, vect_getter
+from baseprocessor import (
+    HDF5Container,
+    Rule,
+    norm_getter,
+    simple_getter,
+    vect_getter,
+    oct_vect_getter,
+)


 # Getters
@@ -1706,6 +1713,7 @@ class PostProcessor(HDF5Container):
                        field + dir,
                        partial(vect_getter, field, i),
                        self.unit_key[field],
+                        oct_getter=partial(oct_vect_getter, field, i),
                    )

                # Radial