@blee1234 As discussed offline, this is the ticket tracking the TODOs in https://github.com/linkedin/feathr/pull/309/ and refactoring needed to merge with compute model branch.