fbpcs/private_computation/stage_flows/private_computation_local_test_stage_flow.py [99:120]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        elif self is self.PREPARE:
            return PrepareDataStageService(
                args.onedocker_svc,
                args.onedocker_binary_config_map,
                update_status_to_complete=True,
            )
        elif self is self.COMPUTE:
            return ComputeMetricsStageService(
                args.onedocker_binary_config_map,
                args.mpc_svc,
            )
        elif self is self.AGGREGATE:
            return AggregateShardsStageService(
                args.onedocker_binary_config_map,
                args.mpc_svc,
            )
        elif self is self.POST_PROCESSING_HANDLERS:
            return PostProcessingStageService(
                args.storage_svc, args.post_processing_handlers
            )
        else:
            raise NotImplementedError(f"No stage service configured for {self}")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



fbpcs/private_computation/stage_flows/private_computation_stage_flow.py [147:168]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        elif self is self.PREPARE:
            return PrepareDataStageService(
                args.onedocker_svc,
                args.onedocker_binary_config_map,
                update_status_to_complete=True,
            )
        elif self is self.COMPUTE:
            return ComputeMetricsStageService(
                args.onedocker_binary_config_map,
                args.mpc_svc,
            )
        elif self is self.AGGREGATE:
            return AggregateShardsStageService(
                args.onedocker_binary_config_map,
                args.mpc_svc,
            )
        elif self is self.POST_PROCESSING_HANDLERS:
            return PostProcessingStageService(
                args.storage_svc, args.post_processing_handlers
            )
        else:
            raise NotImplementedError(f"No stage service configured for {self}")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -