(self)
| 82 | ) |
| 83 | |
| 84 | def test_script(self): |
| 85 | input_param, input_shape, expected_shape = TEST_CASE_MILMODEL[0] |
| 86 | net = MILModel(**input_param) |
| 87 | test_data = torch.randn(input_shape, dtype=torch.float) |
| 88 | test_script_save(net, test_data) |
| 89 | |
| 90 | |
| 91 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected