Skip to content

Commit 63b8732

Browse files
committed
Fix test
Signed-off-by: Nick Quinn <nicholas_quinn@apple.com>
1 parent 50cf399 commit 63b8732

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

sdk/python/feast/types.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
"UNIX_TIMESTAMP": "UNIX_TIMESTAMP",
3838
"MAP": "MAP",
3939
"JSON": "JSON",
40+
"SCALAR_MAP": "SCALAR_MAP",
4041
}
4142

4243

@@ -93,6 +94,7 @@ class PrimitiveFeastType(Enum):
9394
UUID = 13
9495
TIME_UUID = 14
9596
DECIMAL = 15
97+
SCALAR_MAP = 16
9698

9799
def to_value_type(self) -> ValueType:
98100
"""
@@ -130,6 +132,7 @@ def __hash__(self):
130132
Uuid = PrimitiveFeastType.UUID
131133
TimeUuid = PrimitiveFeastType.TIME_UUID
132134
Decimal = PrimitiveFeastType.DECIMAL
135+
ScalarMap = PrimitiveFeastType.SCALAR_MAP
133136

134137
SUPPORTED_BASE_TYPES = [
135138
Invalid,
@@ -167,6 +170,7 @@ def __hash__(self):
167170
"UUID": "Uuid",
168171
"TIME_UUID": "TimeUuid",
169172
"DECIMAL": "Decimal",
173+
"SCALAR_MAP": "ScalarMap",
170174
}
171175

172176

@@ -346,6 +350,7 @@ def __hash__(self):
346350
ValueType.DECIMAL: Decimal,
347351
ValueType.DECIMAL_LIST: Array(Decimal),
348352
ValueType.DECIMAL_SET: Set(Decimal),
353+
ValueType.SCALAR_MAP: ScalarMap,
349354
}
350355

351356
FEAST_TYPES_TO_PYARROW_TYPES = {

0 commit comments

Comments
 (0)