def start()

in src/vw-serving/src/vw_serving/sagemaker/utils.py [0:0]

• 9 lines of code
• 6 McCabe index (conditional complexity)

    def start(self):
        """
        starts the scheduler, server & workers

        :return:
        """
        if self._num_instances < 1:
            raise ValueError("Invalid learner instance count")

        # Run the parameter server scheduler and server processes. Only
        # one of the peers will run the scheduler process.
        if self._my_dns == self._scheduler_dns:
            self.scheduler_pid = self._run_parameter_server(role="scheduler")

        # current container is a server container
        if self._my_dns in self._get_server_dns_list():
            self.server_pid = self._run_parameter_server(role="server")

        # setup the env vars for the current process to run as a worker
        os.environ.update(self._get_dmlc_envs("worker"))
        logging.info("Environment: %s", os.environ.copy())