def get_distribution_hosts()

in src/sagemaker_training/environment.py [0:0]

• 10 lines of code
• 4 McCabe index (conditional complexity)

    def get_distribution_hosts(self):
        """
        Get the list of all hosts in all distribution instance groups
        """
        distribution_hosts = []
        instance_groups_config = self._resource_config.get("instance_groups", [])
        if instance_groups_config:
            for group in instance_groups_config:
                if group["instance_group_name"] in self._distribution_instance_groups:
                    distribution_hosts.extend(group["hosts"])
        else:
            # local mode
            distribution_hosts = self.hosts.copy()
        return distribution_hosts