Bug/stack to dataset

Federico Izzorequested to merge
bug/stack-to-dataset into epic/neurips

Fix stack to dataset and data loading

Merge request reports