@@ -55,8 +55,8 @@ def upgrade_generators_yaml(problem_path: Path, bar: ProgressBar) -> None:
55
55
generators_yaml = problem_path / "generators" / "generators.yaml"
56
56
if not generators_yaml .is_file ():
57
57
return
58
- data = read_yaml (generators_yaml )
59
- if data is None or not isinstance (data , dict ):
58
+ yaml_data = read_yaml (generators_yaml )
59
+ if yaml_data is None or not isinstance (yaml_data , dict ):
60
60
return
61
61
62
62
changed = False
@@ -67,17 +67,19 @@ def upgrade_generators_yaml(problem_path: Path, bar: ProgressBar) -> None:
67
67
("invalid_outputs" , "invalid_output" ),
68
68
("valid_outputs" , "valid_output" ),
69
69
]
70
- for old_name , new_name in rename :
71
- if old_name in data :
72
- if new_name in data :
73
- bar .error (
74
- f"can't rename 'data.{ old_name } ', 'data.{ new_name } ' already exists in generators.yaml" ,
75
- resume = True ,
76
- )
77
- continue
78
- bar .log (f"renaming 'data.{ old_name } ' to 'data.{ new_name } ' in generators.yaml" )
79
- ryaml_replace (data , old_name , new_name )
80
- changed = True
70
+ if "data" in yaml_data and isinstance (yaml_data ["data" ], dict ):
71
+ data = yaml_data ["data" ]
72
+ for old_name , new_name in rename :
73
+ if old_name in data :
74
+ if new_name in data :
75
+ bar .error (
76
+ f"can't rename 'data.{ old_name } ', 'data.{ new_name } ' already exists in generators.yaml" ,
77
+ resume = True ,
78
+ )
79
+ continue
80
+ bar .log (f"renaming 'data.{ old_name } ' to 'data.{ new_name } ' in generators.yaml" )
81
+ ryaml_replace (data , old_name , new_name )
82
+ changed = True
81
83
82
84
def upgrade_generated_testdata_yaml (data : dict [str , Any ], path : str ) -> bool :
83
85
changed = False
@@ -111,10 +113,10 @@ def upgrade_generated_testdata_yaml(data: dict[str, Any], path: str) -> bool:
111
113
)
112
114
return changed
113
115
114
- changed |= upgrade_generated_testdata_yaml (data , "" )
116
+ changed |= upgrade_generated_testdata_yaml (yaml_data , "" )
115
117
116
118
if changed :
117
- write_yaml (data , generators_yaml )
119
+ write_yaml (yaml_data , generators_yaml )
118
120
119
121
120
122
def upgrade_statement (problem_path : Path , bar : ProgressBar ) -> None :
0 commit comments