Skip to content

Commit 152d7cf

Browse files
adds support for more athena supported types and comparators
1 parent 35d3952 commit 152d7cf

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/sagemaker/feature_store/dataset_builder.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,28 +47,30 @@ class TableType(Enum):
4747
class JoinTypeEnum(Enum):
4848
"""Enum of Join types.
4949
50-
The Join comparator can be "INNER_JOIN", "LEFT_JOIN", "RIGHT_JOIN", "FULL_JOIN"
50+
The Join type can be "INNER_JOIN", "LEFT_JOIN", "RIGHT_JOIN", "FULL_JOIN", or "CROSS_JOIN".
5151
"""
5252

5353
INNER_JOIN = "JOIN"
5454
LEFT_JOIN = "LEFT JOIN"
5555
RIGHT_JOIN = "RIGHT JOIN"
5656
FULL_JOIN = "FULL JOIN"
57+
CROSS_JOIN = "CROSS JOIN"
5758

5859

5960
@attr.s
6061
class JoinComparatorEnum(Enum):
6162
"""Enum of Join comparators.
6263
6364
The Join comparator can be "EQUALS", "GREATER_THAN", "LESS_THAN",
64-
"GREATER_THAN_OR_EQUAL_TO", or "LESS_THAN_OR_EQUAL_TO"
65+
"GREATER_THAN_OR_EQUAL_TO", "LESS_THAN_OR_EQUAL_TO" or "NOT_EQUAL_TO"
6566
"""
6667

6768
EQUALS = "="
6869
GREATER_THAN = ">"
6970
GREATER_THAN_OR_EQUAL_TO = ">="
7071
LESS_THAN = "<"
7172
LESS_THAN_OR_EQUAL_TO = "<="
73+
NOT_EQUAL_TO = "<>"
7274

7375

7476
@attr.s

0 commit comments

Comments
 (0)