Exporting Custom Databunch

I’m trying to export my learner using learn.export() but the issue is that the databunch that is used is customized and I’m getting the error AttributeError: 'ImageCDataset' object has no attribute 'get_state'. If I put in a get_state method in ImageCDataset will it successfully export or will I need to do more than that? Is this going to be trivial or should I look for a different solution?