Skip to content

Commit

Permalink
chore: release 0.30.2 (#63)
Browse files Browse the repository at this point in the history
Signed-off-by: ThibaultFy <[email protected]>
  • Loading branch information
ThibaultFy authored Dec 13, 2022
1 parent f8c6f32 commit dc46f45
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 2 deletions.
2 changes: 1 addition & 1 deletion substrafl/__version__.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "0.30.1"
__version__ = "0.30.2"
8 changes: 7 additions & 1 deletion substrafl/experiment.py
Original file line number Diff line number Diff line change
Expand Up @@ -224,6 +224,7 @@ def execute_experiment(
clean_models: bool = True,
name: Optional[str] = None,
additional_metadata: Optional[Dict] = None,
task_submission_batch_size: int = 500,
) -> substra.sdk.models.ComputePlan:
"""Run a complete experiment. This will train (on the `train_data_nodes`) and test (on the
`test_data_nodes`) your `algo` with the specified `strategy` `n_rounds` times and return the
Expand Down Expand Up @@ -263,7 +264,10 @@ def execute_experiment(
quickly so should be set to True unless needed. Defaults to True.
name (str, Optional): Optional name chosen by the user to identify the compute plan. If None,
the compute plan name is set to the timestamp.
additional_metadata(dict, Optional): Optional dictionary of metadata to be passed to the Substra WebApp.
additional_metadata(dict, typing.Optional): Optional dictionary of metadata to be passed to the Substra WebApp.
task_submission_batch_size(int): The compute plan tasks are submitted by batch. The higher the batch size,
the faster the submission, a batch size that is too high makes the submission fail.
Rule of thumb: batch_size = math.floor(120000 / number_of_samples_per_task)
Returns:
ComputePlan: The generated compute plan
Expand Down Expand Up @@ -365,6 +369,8 @@ def execute_experiment(
name=name or timestamp,
metadata=cp_metadata,
),
auto_batching=True,
batch_size=task_submission_batch_size,
)
logger.info(("The compute plan has been registered to Substra, its key is {0}.").format(compute_plan.key))
return compute_plan

0 comments on commit dc46f45

Please sign in to comment.