FilesExpand file tree

enable-model-init

/

saved_datasets.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (46 loc) · 1.38 KB

enable-model-init

/

saved_datasets.py

Copy path

Top

File metadata and controls

• Code
• Blame

57 lines (46 loc) · 1.38 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

import click

import yaml

from feast import utils

from feast.cli.cli_options import tagsOption

from feast.errors import FeastObjectNotFoundException

from feast.repo_operations import create_feature_store

@click.group(name="saved-datasets")

def saved_datasets_cmd():

"""

[Experimental] Access saved datasets

"""

pass

@saved_datasets_cmd.command("describe")

@click.argument("name", type=click.STRING)

@click.pass_context

def saved_datasets_describe(ctx: click.Context, name: str):

"""

[Experimental] Describe a saved dataset

"""

store = create_feature_store(ctx)

try:

saved_dataset = store.get_saved_dataset(name)

except FeastObjectNotFoundException as e:

print(e)

exit(1)

print(

yaml.dump(

yaml.safe_load(str(saved_dataset)),

default_flow_style=False,

sort_keys=False,

)

)

@saved_datasets_cmd.command(name="list")

@tagsOption

@click.pass_context

def saved_datasets_list(ctx: click.Context, tags: list[str]):

"""

[Experimental] List all saved datasets

"""

store = create_feature_store(ctx)

table = []

tags_filter = utils.tags_list_to_dict(tags)

for saved_dataset in store.list_saved_datasets(tags=tags_filter):

table.append([saved_dataset.name])

from tabulate import tabulate

print(tabulate(table, headers=["NAME"], tablefmt="plain"))