From 27afa6c39b23e28a1788ec5c52c6260d04bfb5b1 Mon Sep 17 00:00:00 2001 From: Herman Schaaf Date: Tue, 1 Aug 2023 15:52:03 +0100 Subject: [PATCH] Fix JSON type handling --- cloudquery/sdk/scalar/scalar_factory.py | 3 ++- cloudquery/sdk/scalar/string.py | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cloudquery/sdk/scalar/scalar_factory.py b/cloudquery/sdk/scalar/scalar_factory.py index 6eabaca..8bbf1d4 100644 --- a/cloudquery/sdk/scalar/scalar_factory.py +++ b/cloudquery/sdk/scalar/scalar_factory.py @@ -7,6 +7,7 @@ from .date64 import Date64 from .float import Float from .int import Int +from .json import JSON from .list import List from .scalar import ScalarInvalidTypeError from .string import String @@ -89,6 +90,6 @@ def new_scalar(self, dt: pa.DataType): elif dt == UUIDType(): return UUID() elif dt == JSONType(): - return String() + return JSON() else: raise ScalarInvalidTypeError("Invalid type {} for scalar".format(dt)) diff --git a/cloudquery/sdk/scalar/string.py b/cloudquery/sdk/scalar/string.py index f9bbc26..9fd3a79 100644 --- a/cloudquery/sdk/scalar/string.py +++ b/cloudquery/sdk/scalar/string.py @@ -1,5 +1,4 @@ from cloudquery.sdk.scalar import Scalar, ScalarInvalidTypeError -from .scalar import NULL_VALUE class String(Scalar):