(self, key_in_ckpt, use_trace)
| 49 | |
| 50 | @parameterized.expand([TEST_CASE_1, TEST_CASE_2, TEST_CASE_3]) |
| 51 | def test_export(self, key_in_ckpt, use_trace): |
| 52 | meta_file = os.path.join(TESTS_PATH, "testing_data", "metadata.json") |
| 53 | config_file = os.path.join(TESTS_PATH, "testing_data", "inference.json") |
| 54 | with tempfile.TemporaryDirectory() as tempdir: |
| 55 | def_args = {"meta_file": "will be replaced by `meta_file` arg"} |
| 56 | def_args_file = os.path.join(tempdir, "def_args.yaml") |
| 57 | |
| 58 | ckpt_file = os.path.join(tempdir, "model.pt") |
| 59 | ts_file = os.path.join(tempdir, "model.ts") |
| 60 | |
| 61 | parser = ConfigParser() |
| 62 | parser.export_config_file(config=def_args, filepath=def_args_file) |
| 63 | parser.read_config(config_file) |
| 64 | net = parser.get_parsed_content("network_def") |
| 65 | save_state(src=net if key_in_ckpt == "" else {key_in_ckpt: net}, path=ckpt_file) |
| 66 | |
| 67 | cmd = ["coverage", "run", "-m", "monai.bundle", "ckpt_export", "network_def", "--filepath", ts_file] |
| 68 | cmd += ["--meta_file", meta_file, "--config_file", f"['{config_file}','{def_args_file}']", "--ckpt_file"] |
| 69 | cmd += [ckpt_file, "--key_in_ckpt", key_in_ckpt, "--args_file", def_args_file] |
| 70 | if use_trace == "True": |
| 71 | cmd += ["--use_trace", use_trace, "--input_shape", "[1, 1, 96, 96, 96]"] |
| 72 | command_line_tests(cmd) |
| 73 | self.assertTrue(os.path.exists(ts_file)) |
| 74 | |
| 75 | _, metadata, extra_files = load_net_with_metadata( |
| 76 | ts_file, more_extra_files=["inference.json", "def_args.json"] |
| 77 | ) |
| 78 | self.assertIn("schema", metadata) |
| 79 | self.assertIn("meta_file", json.loads(extra_files["def_args.json"])) |
| 80 | self.assertIn("network_def", json.loads(extra_files["inference.json"])) |
| 81 | |
| 82 | @parameterized.expand([TEST_CASE_1, TEST_CASE_2, TEST_CASE_3]) |
| 83 | def test_default_value(self, key_in_ckpt, use_trace): |
nothing calls this directly
no test coverage detected