def get_instance_groups()

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

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

    def get_instance_groups(self):  # type: () -> list
        """
        List of instance groups provided for the job
        """
        instance_groups = []
        instance_groups_config = self._resource_config.get("instance_groups", [])
        # log missing instance groups and return empty list
        if not instance_groups_config:
            logger.info("instance_groups entry not present in resource_config")

        for group in instance_groups_config:
            instance_groups.append(group["instance_group_name"])
        return instance_groups