Visitar URL original
Fixed linter issue · feast-dev/feast@5c0e1f5 · GitHub
Skip to content

Commit 5c0e1f5

Browse files
committed
Fixed linter issue
Signed-off-by: Nick Quinn <nicholas_quinn@apple.com>
1 parent 90cd101 commit 5c0e1f5

1 file changed

Lines changed: 20 additions & 5 deletions

File tree

‎sdk/python/tests/unit/test_type_map.py‎

Lines changed: 20 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
16891704
class TestNestedCollectionTypes:
16901705
"""Tests for nested collection type proto conversion (VALUE_LIST, VALUE_SET)."""
16911706

0 commit comments

Comments
 (0)