Matt Kidd

**Update** I have fixed the error you can see at the bottom! **Original message** Hi there, thanks for the update, this doesn't work for me however: I get the following...

@fadi212 @abdksyed @avinashok In the above thread someone has suggested a solution with a [working colab example that you can use](https://colab.research.google.com/drive/16wqA3oTUf7yzUKsSSZxiMf1443_ZO3wC). They are also fixing up and adding this class...

@quasimik Great work! Could you provide a step by step of how we use your new class `LayoutLmv2RelationExtractionDecoder`? [I see you have added to this part here](https://github.com/woflowinc/transformers/blob/2a8b0da433f8cbe62809ab13da7197b5c419429d/src/transformers/models/layoutlmv2/modeling_layoutlmv2.py#L558)

I am also receiving this error on Sagemaker but not locally, I have noticed that this occurs when the `.dataset/` folder does not contain a single file like: `dataset.arrow` but...