diff --git a/dlrover/python/master/resource/job.py b/dlrover/python/master/resource/job.py index 890ca7572..8a317657a 100644 --- a/dlrover/python/master/resource/job.py +++ b/dlrover/python/master/resource/job.py @@ -232,7 +232,7 @@ def _init_job_resource_by_optimizer(self): plan = self._resource_optimizer.generate_opt_plan(self._job_stage) if not plan or plan.empty(): logger.info("Use the default plan to start the job") - plan = ResourcePlan.new_default_plan() + plan = self._gen_default_resource_plan() self._job_stage = JobOptStage.WORKER_INITIAL if ( @@ -260,6 +260,10 @@ def _init_job_resource_by_optimizer(self): ps_resource.node_resource.memory, ) + def _gen_default_resource_plan(self): + plan = ResourcePlan.new_default_plan() + return plan + def init_job_resource(self, job_resource: JobResource): """Adjust the initial resource of typed pods by EasyDL. Args: