feat(framework) Update context registration when running an app directory #12725

Job	Run time
Build, test and upload wheel	1m 15s
Framework / bare	3m 46s
Strategy / FedMedian	1m 20s
Framework / bare-https	1m 20s
Strategy / FedTrimmedAvg	1m 10s
Framework / bare-client-auth	59s
Strategy / QFedAvg	1m 13s
Framework / framework-jax	1m 21s
Strategy / FaultTolerantFedAvg	1m 16s
Framework / framework-pytorch	2m 17s
Strategy / FedAvgM	1m 18s
Framework / framework-tensorflow	3m 9s
Strategy / FedAdam	1m 16s
Framework / framework-opacus	2m 18s
Strategy / FedAdagrad	1m 12s
Framework / framework-pytorch-lightning	2m 0s
Strategy / FedYogi	1m 21s
Framework / framework-scikit-learn	1m 45s
Framework / framework-fastai	3m 17s
Framework / framework-pandas	1m 13s
	34m 46s

Provide feedback