Method test_ill_args

tests/networks/nets/test_milmodel.py:74–82

(self)

            self.assertEqual(result.shape, expected_shape)

    def test_ill_args(self):
        with self.assertRaises(ValueError):
            MILModel(
                num_classes=5,
                pretrained=False,
                backbone="resnet50",
                backbone_num_features=2048,
                mil_mode="att_trans_pyramid",
            )

    def test_script(self):
        input_param, input_shape, expected_shape = TEST_CASE_MILMODEL[0]