Skip to content

Commit

Permalink
chore: fix enum
Browse files Browse the repository at this point in the history
Signed-off-by: ThibaultFy <[email protected]>
  • Loading branch information
ThibaultFy committed Sep 29, 2023
1 parent 86e5078 commit 46ff77e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions benchmark/camelyon/pure_substrafl/register_assets.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ def add_duplicated_dataset(
data_sample_folder: os.PathLike,
asset_keys: dict,
msp_id: str,
kind: str = TaskType.TRAIN,
kind: str = TaskType.TRAIN.value,
) -> dict:
"""Update asset_keys.msp_id.{kind}_data_sample_keys so there is exactly `nb_data_sample` keys
by adding the `data_sample_folder` as a data sample with the provided `client` as many times as it's need
Expand Down Expand Up @@ -126,7 +126,7 @@ def add_duplicated_dataset(
}
msp_id (str): asset_keys key where to find the registered assets for the given client
kind (str, optional): Kind of data sample to add, either train or test. Defaults to
TaskType.TRAIN.
TaskType.TRAIN.value.
Returns:
dict: The updated asset_keys.
Expand Down Expand Up @@ -193,7 +193,7 @@ def get_train_data_nodes(
data_sample_folder=train_folder,
asset_keys=asset_keys,
msp_id=msp_id,
kind=TaskType.TRAIN,
kind=TaskType.TRAIN.value,
)

train_data_nodes.append(
Expand Down

0 comments on commit 46ff77e

Please sign in to comment.