diff --git a/src/cloudai/_core/test_scenario_parser.py b/src/cloudai/_core/test_scenario_parser.py index ba00119a..9d20e3a3 100644 --- a/src/cloudai/_core/test_scenario_parser.py +++ b/src/cloudai/_core/test_scenario_parser.py @@ -129,11 +129,30 @@ def _create_section_test(self, section: str, test_info: Dict[str, Any]) -> Test: f"reference from the test scenario schema." ) - test = copy.deepcopy(self.test_mapping[test_name]) - test.test_template = self.test_mapping[test_name].test_template + original_test = self.test_mapping[test_name] + + test = Test( + name=original_test.name, + description=original_test.description, + test_template=original_test.test_template, + env_vars=copy.deepcopy(original_test.env_vars), + cmd_args=copy.deepcopy(original_test.cmd_args), + extra_env_vars=copy.deepcopy(original_test.extra_env_vars), + extra_cmd_args=original_test.extra_cmd_args, + dependencies=copy.deepcopy(original_test.dependencies), + iterations=original_test.iterations, + num_nodes=original_test.num_nodes, + nodes=original_test.nodes, + sol=original_test.sol, + weight=original_test.weight, + ideal_perf=original_test.ideal_perf, + time_limit=original_test.time_limit, + ) + test.section_name = section test.num_nodes = int(test_info.get("num_nodes", 1)) test.nodes = test_info.get("nodes", []) + return test def _parse_dependencies_for_test(