def _get_instance_count_ref()

in src/sagemaker_core/tools/resources_codegen.py [0:0]

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

    def _get_instance_count_ref(self, resource_name: str) -> str:
        """Get the instance count reference for a resource object.
        Args:
            resource_name (str): The resource name.
        Returns:
            str: The instance count reference for resource object
        """

        if resource_name == "TrainingJob":
            return """(
                sum(instance_group.instance_count for instance_group in self.resource_config.instance_groups)
                if self.resource_config.instance_groups and not isinstance(self.resource_config.instance_groups, Unassigned)
                else self.resource_config.instance_count
            )
            """
        elif resource_name == "TransformJob":
            return "self.transform_resources.instance_count"
        elif resource_name == "ProcessingJob":
            return "self.processing_resources.cluster_config.instance_count"

        raise ValueError(f"Instance count reference not found for resource {resource_name}")