Skip to content

Commit b7e33ee

Browse files
committed
tests for pg and mysql
Signed-off-by: Achal Shah <[email protected]>
1 parent 9add91d commit b7e33ee

File tree

3 files changed

+310
-12
lines changed

3 files changed

+310
-12
lines changed

sdk/python/feast/infra/registry_stores/sql.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -53,23 +53,23 @@
5353
entities = Table(
5454
"entities",
5555
metadata,
56-
Column("entity_id", String, primary_key=True),
56+
Column("entity_id", String(50), primary_key=True),
5757
Column("last_updated_timestamp", BigInteger, nullable=False),
5858
Column("entity_proto", LargeBinary, nullable=False),
5959
)
6060

6161
data_sources = Table(
6262
"data_sources",
6363
metadata,
64-
Column("data_source_name", String, primary_key=True),
64+
Column("data_source_name", String(50), primary_key=True),
6565
Column("last_updated_timestamp", BigInteger, nullable=False),
6666
Column("data_source_proto", LargeBinary, nullable=False),
6767
)
6868

6969
feature_views = Table(
7070
"feature_views",
7171
metadata,
72-
Column("feature_view_name", String, primary_key=True),
72+
Column("feature_view_name", String(50), primary_key=True),
7373
Column("last_updated_timestamp", BigInteger, nullable=False),
7474
Column("materialized_intervals", LargeBinary, nullable=True),
7575
Column("feature_view_proto", LargeBinary, nullable=False),
@@ -78,47 +78,47 @@
7878
request_feature_views = Table(
7979
"request_feature_views",
8080
metadata,
81-
Column("feature_view_name", String, primary_key=True),
81+
Column("feature_view_name", String(50), primary_key=True),
8282
Column("last_updated_timestamp", BigInteger, nullable=False),
8383
Column("feature_view_proto", LargeBinary, nullable=False),
8484
)
8585

8686
on_demand_feature_views = Table(
8787
"on_demand_feature_views",
8888
metadata,
89-
Column("feature_view_name", String, primary_key=True),
89+
Column("feature_view_name", String(50), primary_key=True),
9090
Column("last_updated_timestamp", BigInteger, nullable=False),
9191
Column("feature_view_proto", LargeBinary, nullable=False),
9292
)
9393

9494
feature_user_metadata = Table(
9595
"feature_metadata",
9696
metadata,
97-
Column("feature_name", String, primary_key=True),
97+
Column("feature_name", String(50), primary_key=True),
9898
Column("last_updated_timestamp", BigInteger, nullable=False),
9999
Column("feature_metadata_binary", LargeBinary, nullable=False),
100100
)
101101

102102
feature_services = Table(
103103
"feature_services",
104104
metadata,
105-
Column("feature_service_name", String, primary_key=True),
105+
Column("feature_service_name", String(50), primary_key=True),
106106
Column("last_updated_timestamp", BigInteger, nullable=False),
107107
Column("feature_service_proto", LargeBinary, nullable=False),
108108
)
109109

110110
saved_datasets = Table(
111111
"saved_datasets",
112112
metadata,
113-
Column("saved_dataset_name", String, primary_key=True),
113+
Column("saved_dataset_name", String(50), primary_key=True),
114114
Column("last_updated_timestamp", BigInteger, nullable=False),
115115
Column("saved_dataset_proto", LargeBinary, nullable=False),
116116
)
117117

118118
validation_references = Table(
119119
"validation_references",
120120
metadata,
121-
Column("validation_reference_name", String, primary_key=True),
121+
Column("validation_reference_name", String(50), primary_key=True),
122122
Column("last_updated_timestamp", BigInteger, nullable=False),
123123
Column("validation_reference_proto", LargeBinary, nullable=False),
124124
)

0 commit comments

Comments
 (0)