def inference()

in MMS/dicom_featurization_service.py [0:0]

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

    def inference(self, img):
       # image featurization
       with torch.no_grad():
           output = self.model(img)
       
       return output.numpy()[0]