@@ -1620,7 +1620,11 @@ def test_decimal_object_list_roundtrip(self):
16201620 """List of decimal.Decimal objects -> proto -> list of decimal.Decimal roundtrip."""
16211621 import decimal
16221622
1623- vals = [decimal .Decimal ("10.5" ), decimal .Decimal ("20.75" ), decimal .Decimal ("30.125" )]
1623+ vals = [
1624+ decimal .Decimal ("10.5" ),
1625+ decimal .Decimal ("20.75" ),
1626+ decimal .Decimal ("30.125" ),
1627+ ]
16241628 protos = python_values_to_proto_values ([vals ], ValueType .DECIMAL_LIST )
16251629 result = feast_value_type_to_python_type (protos [0 ])
16261630 assert all (isinstance (r , decimal .Decimal ) for r in result )
@@ -1663,8 +1667,14 @@ def test_decimal_value_type_string_conversion(self):
16631667 from feast .type_map import _convert_value_type_str_to_value_type
16641668
16651669 assert _convert_value_type_str_to_value_type ("DECIMAL" ) == ValueType .DECIMAL
1666- assert _convert_value_type_str_to_value_type ("DECIMAL_LIST" ) == ValueType .DECIMAL_LIST
1667- assert _convert_value_type_str_to_value_type ("DECIMAL_SET" ) == ValueType .DECIMAL_SET
1670+ assert (
1671+ _convert_value_type_str_to_value_type ("DECIMAL_LIST" )
1672+ == ValueType .DECIMAL_LIST
1673+ )
1674+ assert (
1675+ _convert_value_type_str_to_value_type ("DECIMAL_SET" )
1676+ == ValueType .DECIMAL_SET
1677+ )
16681678
16691679 def test_decimal_proto_field_name_mapping (self ):
16701680 """PROTO_VALUE_TO_VALUE_TYPE_MAP and VALUE_TYPE_TO_PROTO_VALUE_MAP include DECIMAL."""
@@ -1674,10 +1684,14 @@ def test_decimal_proto_field_name_mapping(self):
16741684 )
16751685
16761686 assert PROTO_VALUE_TO_VALUE_TYPE_MAP ["decimal_val" ] == ValueType .DECIMAL
1677- assert PROTO_VALUE_TO_VALUE_TYPE_MAP ["decimal_list_val" ] == ValueType .DECIMAL_LIST
1687+ assert (
1688+ PROTO_VALUE_TO_VALUE_TYPE_MAP ["decimal_list_val" ] == ValueType .DECIMAL_LIST
1689+ )
16781690 assert PROTO_VALUE_TO_VALUE_TYPE_MAP ["decimal_set_val" ] == ValueType .DECIMAL_SET
16791691 assert VALUE_TYPE_TO_PROTO_VALUE_MAP [ValueType .DECIMAL ] == "decimal_val"
1680- assert VALUE_TYPE_TO_PROTO_VALUE_MAP [ValueType .DECIMAL_LIST ] == "decimal_list_val"
1692+ assert (
1693+ VALUE_TYPE_TO_PROTO_VALUE_MAP [ValueType .DECIMAL_LIST ] == "decimal_list_val"
1694+ )
16811695 assert VALUE_TYPE_TO_PROTO_VALUE_MAP [ValueType .DECIMAL_SET ] == "decimal_set_val"
16821696
16831697 def test_decimal_pandas_type (self ):
@@ -1686,6 +1700,7 @@ def test_decimal_pandas_type(self):
16861700
16871701 assert feast_value_type_to_pandas_type (ValueType .DECIMAL ) == "object"
16881702
1703+
16891704class TestNestedCollectionTypes :
16901705 """Tests for nested collection type proto conversion (VALUE_LIST, VALUE_SET)."""
16911706
0 commit comments