E2E

Fix divide_dataset in Federated Datasets #8747

Job	Run time
Build, test and upload wheel	1m 24s
Strategy / FedMedian	1m 24s
Strategy / FedTrimmedAvg	1m 14s
Strategy / QFedAvg	1m 12s
Framework / bare	2m 55s
Strategy / FaultTolerantFedAvg	1m 12s
Framework / bare-https	1m 20s
Strategy / FedAvgM	1m 19s
Framework / jax	2m 55s
Strategy / FedAdam	1m 13s
Framework / pytorch	2m 45s
Strategy / FedAdagrad	2m 1s
Framework / tensorflow	3m 3s
Strategy / FedYogi	1m 10s
Framework / tabnet	2m 50s
Framework / opacus	3m 17s
Framework / pytorch-lightning	2m 38s
Framework / scikit-learn	2m 39s
Framework / fastai	4m 16s
Framework / pandas	1m 47s
	42m 34s

Provide feedback