Pushed some changes to https://github.com/flyteorg/flytekit/pull/1240 and made some comments as im struggling a bit with the design (and getting the tests to pass). ☹️

Oh sorry about that! I haven’t taken a look at it yet as I’ve been busy this week. Let me review it the next week and get back to you.

@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.