Skip to content

Commit

Permalink
update DatasetV2 to have all old attributes and update migrations
Browse files Browse the repository at this point in the history
  • Loading branch information
BrendanSchell committed Jul 4, 2024
1 parent 447d903 commit d3dc366
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions packages/syft/src/syft/service/dataset/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -474,6 +474,18 @@ class DatasetV2(SyftObject):
created_at: DateTime = DateTime.now()
uploader: Contributor

__attr_searchable__ = [
"name",
"citation",
"url",
"description",
"action_ids",
"summary",
]
__attr_unique__ = ["name"]
__repr_attrs__ = ["name", "url", "created_at"]
__table_sort_attr__ = "Created at"


@serializable()
class Dataset(SyftObject):
Expand Down Expand Up @@ -922,27 +934,35 @@ def createdataset_to_dataset() -> list[Callable]:
def migrate_dataset_v2_to_v3() -> list[Callable]:
return [
make_set_default("summary", None),
drop("__repr_attrs__"),
make_set_default("__repr_attrs__", ["name", "summary", "url", "created_at"]),
]


@migrate(Dataset, DatasetV2)
def migrate_dataset_v3_to_v2() -> list[Callable]:
return [
drop("summary"),
drop("__repr_attrs__"),
make_set_default("__repr_attrs__", ["name", "url", "created_at"]),
]


@migrate(CreateDatasetV2, CreateDataset)
def migrate_create_dataset_v2_to_v3() -> list[Callable]:
return [
make_set_default("summary", None),
drop("__repr_attrs__"),
make_set_default("__repr_attrs__", ["name", "summary", "url"]),
]


@migrate(CreateDataset, CreateDatasetV2)
def migrate_create_dataset_v3_to_v2() -> list[Callable]:
return [
drop("summary"),
drop("__repr_attrs__"),
make_set_default("__repr_attrs__", ["name", "url"]),
]


Expand Down

0 comments on commit d3dc366

Please sign in to comment.