Rikai looks really interesting, how about integrating with flyte.org

[kumare3]

Flyte supports custom types, that users can use natively in their code and Flyte handles the transport and maintenance of the actual data. This alleviates the complexity of using data types. Flyte is extremely type-safe and automatically provides things like memoization etc for user data.

Example of writing custom types can be found here

Some already implemented types can be seen here Check Pander, Modin, Spark, GreatExpectations etc