Stricter type checking on FloatDataType return type