@tall-lock-23197 Just pushed some changes and redesigned the transformers as suggested. However still getting an error in the test , the issue is when executing the task and annotating the return as TFRecordFile or TFRecordDirectory , it thinks its ABCMeta metaclass. Ive added comments in PR.