Update medical_pipeline.py

medical_pipeline.py

@@ -152,6 +152,6 @@ class MedicalPipeline:
         elif keys == 'KiTS2019':
             label = self.map_to_classes(label, 2)
 
-        return image, label, img_prompt, mask_prompt
+        return image, label