MINOR: Fix Sample Data Deserialization (#15079)

This commit is contained in:
Mayur Singal 2024-02-09 11:25:53 +05:30 committed by GitHub
parent c635cbe44f
commit 96de1f5402
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -11,7 +11,7 @@
"""
Profiler Processor Step
"""
import json
import traceback
from datetime import datetime
from functools import singledispatch
@ -48,7 +48,11 @@ def upload_sample_data(data: TableData, profiler_interface: ProfilerInterface) -
sample_storage_config = profiler_interface.storage_config
if not sample_storage_config:
return
df = pd.DataFrame(data=data.rows, columns=[i.__root__ for i in data.columns])
deserialized_data = json.loads(data.json())
df = pd.DataFrame(
data=deserialized_data.get("rows", []),
columns=[i.__root__ for i in data.columns],
)
pq_buffer = BytesIO()
df.to_parquet(pq_buffer)
object_key = _get_object_key(