fix(csv): Do not coerce persisted data integer columns to float (#20760)

* Replace pd.DataFrame.from_records with pd.DataFrame

* Remove unused code

* Update core.py

* Update core.py

* Update csv.py

* Update core.py
This commit is contained in:
John Bodley 2022-07-19 09:52:47 -07:00 committed by GitHub
parent 8c0ac9017f
commit e1fd90697c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions

View File

@ -2502,8 +2502,13 @@ class Superset(BaseSupersetView): # pylint: disable=too-many-public-methods
obj = _deserialize_results_payload(
payload, query, cast(bool, results_backend_use_msgpack)
)
columns = [c["name"] for c in obj["columns"]]
df = pd.DataFrame.from_records(obj["data"], columns=columns)
df = pd.DataFrame(
data=obj["data"],
dtype=object,
columns=[c["name"] for c in obj["columns"]],
)
logger.info("Using pandas to convert to CSV")
else:
logger.info("Running a query to turn into CSV")