Skip to content

Commit 912fb40

Browse files
achalsachals
committed
plumb
Signed-off-by: Achal Shah <achals@gmail.com>
1 parent 2112bd9 commit 912fb40

File tree

1 file changed

+18
-18
lines changed

1 file changed

+18
-18
lines changed

‎sdk/python/feast/cli.py‎

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,6 @@ def format_options(self, ctx: click.Context, formatter: click.HelpFormatter):
7474
)
7575
@click.option(
7676
"--feature-store-yaml",
77-
"-f",
7877
help="Override the directory where the CLI should look for the feature_store.yaml file.",
7978
)
8079
@click.pass_context
@@ -160,7 +159,7 @@ def ui(ctx: click.Context, host: str, port: int, registry_ttl_sec: int):
160159
repo = ctx.obj["CHDIR"]
161160
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
162161
cli_check_repo(repo, fs_yaml_file)
163-
store = FeatureStore(repo_path=str(repo))
162+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
164163
# Pass in the registry_dump method to get around a circular dependency
165164
store.serve_ui(
166165
host=host,
@@ -179,7 +178,7 @@ def endpoint(ctx: click.Context):
179178
repo = ctx.obj["CHDIR"]
180179
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
181180
cli_check_repo(repo, fs_yaml_file)
182-
store = FeatureStore(repo_path=str(repo))
181+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
183182
endpoint = store.get_feature_server_endpoint()
184183
if endpoint is not None:
185184
_logger.info(
@@ -207,7 +206,7 @@ def data_source_describe(ctx: click.Context, name: str):
207206
repo = ctx.obj["CHDIR"]
208207
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
209208
cli_check_repo(repo, fs_yaml_file)
210-
store = FeatureStore(repo_path=str(repo))
209+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
211210

212211
try:
213212
data_source = store.get_data_source(name)
@@ -236,7 +235,7 @@ def data_source_list(ctx: click.Context):
236235
repo = ctx.obj["CHDIR"]
237236
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
238237
cli_check_repo(repo, fs_yaml_file)
239-
store = FeatureStore(repo_path=str(repo))
238+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
240239
table = []
241240
for datasource in store.list_data_sources():
242241
table.append([datasource.name, datasource.__class__])
@@ -269,7 +268,7 @@ def entity_describe(ctx: click.Context, name: str):
269268
repo = ctx.obj["CHDIR"]
270269
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
271270
cli_check_repo(repo, fs_yaml_file)
272-
store = FeatureStore(repo_path=str(repo))
271+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
273272

274273
try:
275274
entity = store.get_entity(name)
@@ -293,7 +292,7 @@ def entity_list(ctx: click.Context):
293292
repo = ctx.obj["CHDIR"]
294293
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
295294
cli_check_repo(repo, fs_yaml_file)
296-
store = FeatureStore(repo_path=str(repo))
295+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
297296
table = []
298297
for entity in store.list_entities():
299298
table.append([entity.name, entity.description, entity.value_type])
@@ -321,7 +320,7 @@ def feature_service_describe(ctx: click.Context, name: str):
321320
repo = ctx.obj["CHDIR"]
322321
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
323322
cli_check_repo(repo, fs_yaml_file)
324-
store = FeatureStore(repo_path=str(repo))
323+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
325324

326325
try:
327326
feature_service = store.get_feature_service(name)
@@ -347,7 +346,7 @@ def feature_service_list(ctx: click.Context):
347346
repo = ctx.obj["CHDIR"]
348347
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
349348
cli_check_repo(repo, fs_yaml_file)
350-
store = FeatureStore(repo_path=str(repo))
349+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
351350
feature_services = []
352351
for feature_service in store.list_feature_services():
353352
feature_names = []
@@ -380,7 +379,7 @@ def feature_view_describe(ctx: click.Context, name: str):
380379
repo = ctx.obj["CHDIR"]
381380
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
382381
cli_check_repo(repo, fs_yaml_file)
383-
store = FeatureStore(repo_path=str(repo))
382+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
384383

385384
try:
386385
feature_view = store.get_feature_view(name)
@@ -403,8 +402,9 @@ def feature_view_list(ctx: click.Context):
403402
"""
404403
repo = ctx.obj["CHDIR"]
405404
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
405+
406406
cli_check_repo(repo, fs_yaml_file)
407-
store = FeatureStore(repo_path=str(repo))
407+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
408408
table = []
409409
for feature_view in [
410410
*store.list_feature_views(),
@@ -448,7 +448,7 @@ def on_demand_feature_view_describe(ctx: click.Context, name: str):
448448
repo = ctx.obj["CHDIR"]
449449
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
450450
cli_check_repo(repo, fs_yaml_file)
451-
store = FeatureStore(repo_path=str(repo))
451+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
452452

453453
try:
454454
on_demand_feature_view = store.get_on_demand_feature_view(name)
@@ -474,7 +474,7 @@ def on_demand_feature_view_list(ctx: click.Context):
474474
repo = ctx.obj["CHDIR"]
475475
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
476476
cli_check_repo(repo, fs_yaml_file)
477-
store = FeatureStore(repo_path=str(repo))
477+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
478478
table = []
479479
for on_demand_feature_view in store.list_on_demand_feature_views():
480480
table.append([on_demand_feature_view.name])
@@ -579,7 +579,7 @@ def materialize_command(
579579
repo = ctx.obj["CHDIR"]
580580
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
581581
cli_check_repo(repo, fs_yaml_file)
582-
store = FeatureStore(repo_path=str(repo))
582+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
583583
store.materialize(
584584
feature_views=None if not views else views,
585585
start_date=utils.make_tzaware(parser.parse(start_ts)),
@@ -608,7 +608,7 @@ def materialize_incremental_command(ctx: click.Context, end_ts: str, views: List
608608
repo = ctx.obj["CHDIR"]
609609
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
610610
cli_check_repo(repo, fs_yaml_file)
611-
store = FeatureStore(repo_path=str(repo))
611+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
612612
store.materialize_incremental(
613613
feature_views=None if not views else views,
614614
end_date=utils.make_tzaware(datetime.fromisoformat(end_ts)),
@@ -699,7 +699,7 @@ def serve_command(
699699
repo = ctx.obj["CHDIR"]
700700
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
701701
cli_check_repo(repo, fs_yaml_file)
702-
store = FeatureStore(repo_path=str(repo))
702+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
703703

704704
if go:
705705
# Turn on Go feature retrieval.
@@ -722,7 +722,7 @@ def serve_transformations_command(ctx: click.Context, port: int):
722722
repo = ctx.obj["CHDIR"]
723723
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
724724
cli_check_repo(repo, fs_yaml_file)
725-
store = FeatureStore(repo_path=str(repo))
725+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
726726

727727
store.serve_transformations(port)
728728

@@ -762,7 +762,7 @@ def validate(
762762
repo = ctx.obj["CHDIR"]
763763
fs_yaml_file = ctx.obj["FS_YAML_FILE"]
764764
cli_check_repo(repo, fs_yaml_file)
765-
store = FeatureStore(repo_path=str(repo))
765+
store = FeatureStore(repo_path=str(repo), fs_yaml_file=fs_yaml_file)
766766

767767
feature_service = store.get_feature_service(name=feature_service)
768768
reference = store.get_validation_reference(reference)

0 commit comments

Comments
 (0)