Skip to content

Commit 5f08474

Browse files
Fix float inference to use DoubleParameter (64-bit) instead of FloatParameter (32-bit)
Signed-off-by: Shubhambhusate <bhusates6@gmail.com>
1 parent 9fe7356 commit 5f08474

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/databricks/sql/parameters/native.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -659,7 +659,7 @@ def dbsql_parameter_from_primitive(
659659
elif isinstance(value, str):
660660
return StringParameter(value=value, name=name)
661661
elif isinstance(value, float):
662-
return FloatParameter(value=value, name=name)
662+
return DoubleParameter(value=value, name=name)
663663
elif isinstance(value, datetime.datetime):
664664
return TimestampParameter(value=value, name=name)
665665
elif isinstance(value, datetime.date):

tests/unit/test_parameters.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,7 @@ def test_tspark_param_ordinal(self):
295295
(BigIntegerParameter, Primitive.BIGINT),
296296
(BooleanParameter, Primitive.BOOL),
297297
(DateParameter, Primitive.DATE),
298-
(FloatParameter, Primitive.FLOAT),
298+
(DoubleParameter, Primitive.FLOAT),
299299
(VoidParameter, Primitive.NONE),
300300
(TimestampParameter, Primitive.TIMESTAMP),
301301
(MapParameter, Primitive.MAP),
@@ -305,7 +305,7 @@ def test_tspark_param_ordinal(self):
305305
def test_inference(self, _type: TDbsqlParameter, prim: Primitive):
306306
"""This method only tests inferrable types.
307307
308-
Not tested are TinyIntParameter, SmallIntParameter DoubleParameter and TimestampNTZParameter
308+
Not tested are TinyIntParameter, SmallIntParameter, FloatParameter and TimestampNTZParameter
309309
"""
310310

311311
inferred_type = dbsql_parameter_from_primitive(prim.value)

0 commit comments

Comments
 (0)