def _set_capacities()

in analysis/GreenSKU-Framework/src/carbon_model.py [0:0]

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

    def _set_capacities(self) -> None:
        """Set the capacities per socket for the per-socket components."""
        self.capacities = {}
        for key, value in self.data.items():
            if key in self.per_socket:
                if key == 'cpu':
                    self.capacities[key] = value['count'] * value['number']
                elif key == 'memory':
                    self.capacities[key] = value['size'] * value['number']
                elif key == 'cxl':
                    self.capacities[key] = value['size'] * value['number']
                elif key == 'ssd':
                    self.capacities[key] = value['size']  * value['number'] / 1000.0 # convert to TB
                elif key == 'cxl_controller':
                    self.capacities[key] = value['number']
                elif key == 'ssd_reuse':
                    self.capacities[key] = value['size']  * value['number'] / 1000.0 # convert to TB
        
        self.capacities['total_memory'] = self.capacities['memory'] + self.capacities['cxl']