[feature request] Task-based data format implementation

[NISH1001]

We might need to have a data structure specific to certain tasks.
For example, we can do a squad-like input format data structure

---

cc: @xhagrg @muthukumaranR