let trainer's decoratee decorate dataset

domainlab/algos/trainers/a_trainer.py

@@ -244,3 +244,13 @@ def _cal_reg_loss(self, tensor_x, tensor_y, tensor_d, others=None):
         interface for each trainer to implement
         """
         return [], []
+
+    def dset_decoration_args_algo(self, args, ddset):
+        """
+        decorate dataset to get extra entries in load item, for instance,
+        jigen need permutation index
+        this parent class function delegate decoration to its decoratee
+        """
+        if self._decoratee is not None:
+            return self._decoratee.dset_decoration_args_algo(args, ddset)
+        return ddset