def remove_metric()

in src/worker/exporters/custom_exporter.py [0:0]

• 14 lines of code
• 5 McCabe index (conditional complexity)

    def remove_metric(self, field_name):
        """Remove metrics label values"""
        if "gpu_" in field_name:
            gpu_idx = field_name.split('_')[1]
            metric_name = f"{field_name.split('_')[0]}{field_name.split('_')[2]}"
            if int(gpu_idx) < len(GPU_Mapping):
                self.gauges[metric_name].remove(config['gpu_id'][gpu_idx], config['job_id'])
        elif "ib_" in field_name:
            ib_port = field_name.split('_')[1]
            metric_name = f"{field_name.split('_')[0]}{field_name.split('_')[2]}"
            if int(ib_port) < len(IB_Mapping):
                ib_port = field_name.split('_')[1]
            self.gauges[metric_name].remove(config['ib_port'][ib_port], config['job_id'])
        else:
            self.gauges[field_name].remove(config['job_id'])