fix(*:skip) Fix querying dataset size functions #12515

Job	Run time
Build, test and upload wheel	1m 6s
Strategy / FedMedian	1m 10s
Framework / bare	3m 53s
Strategy / FedTrimmedAvg	1m 8s
Framework / bare-https	1m 11s
Strategy / QFedAvg	1m 13s
Framework / bare-client-auth	49s
Strategy / FaultTolerantFedAvg	1m 12s
Framework / framework-jax	1m 12s
Strategy / FedAvgM	1m 9s
Framework / framework-pytorch	1m 55s
Strategy / FedAdam	1m 15s
Framework / framework-tensorflow	3m 5s
Strategy / FedAdagrad	1m 12s
Framework / framework-opacus	2m 3s
Strategy / FedYogi	1m 11s
Framework / framework-pytorch-lightning	1m 33s
Framework / framework-scikit-learn	1m 30s
Framework / framework-fastai	2m 33s
Framework / framework-pandas	1m 9s
	31m 29s

Provide feedback