TypeError: Value passed to parameter 'a' has DataType int64 not in list of allowed values: float16,

NoSuchKey