Prediction function takes an input and returns an output. SuperAlign supports the following data types for the same.

"pandas dataframe" = Pandas DataFrame
"numpy ndarray" = Numpy Ndarray
"text" = Text
"image" = Image
"json" = JSON

These data types can be specified while defining the Input and Output in the Predictor class.