MINOR: Fix sample data upload - binary data error (#15659)

This commit is contained in:
Mayur Singal 2024-03-22 12:13:26 +05:30 committed by GitHub
parent 2208662886
commit ad28af4f4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -17,6 +17,8 @@ from datetime import datetime
from functools import singledispatch
from io import BytesIO
from pydantic.json import ENCODERS_BY_TYPE
from metadata.clients.aws_client import AWSClient
from metadata.generated.schema.entity.data.table import Table, TableData
from metadata.generated.schema.entity.services.connections.connectionBasicType import (
@ -81,6 +83,7 @@ def upload_sample_data(data: TableData, profiler_interface: ProfilerInterface) -
sample_storage_config: DataStorageConfig = profiler_interface.storage_config
if not sample_storage_config:
return
ENCODERS_BY_TYPE[bytes] = lambda v: v.decode("utf-8", "ignore")
deserialized_data = json.loads(data.json())
df = pd.DataFrame(
data=deserialized_data.get("rows", []),