@@ -141,15 +141,14 @@ def _generate_workflow(
141141 template_name : str ,
142142 output_dir : Path ,
143143 max_jobs : int = 250 ,
144- ) -> list :
144+ ) -> int :
145145 """Generate reusable workflow files, returns list of file numbers."""
146146 # GitHub limits workflows to 256 jobs per file. We use a lower default
147147 # (250)
148148 chunks = [
149149 job_datas [index : index + max_jobs ]
150150 for index in range (0 , len (job_datas ), max_jobs )
151151 ]
152- file_numbers = []
153152 env = Environment (
154153 loader = FileSystemLoader (Path (__file__ ).parent .joinpath ("templates" ))
155154 )
@@ -163,13 +162,12 @@ def _generate_workflow(
163162 )
164163 )
165164 yml_file .write ("\n " )
166- file_numbers .append (file_number )
167- return file_numbers
165+ return len (chunks )
168166
169167
170168def generate_test_workflow (
171169 tox_ini_path : Path , workflow_directory_path : Path , operating_systems
172- ) -> list :
170+ ) -> int :
173171 return _generate_workflow (
174172 get_test_job_datas (get_tox_envs (tox_ini_path ), operating_systems ),
175173 "test" ,
@@ -180,7 +178,7 @@ def generate_test_workflow(
180178def generate_lint_workflow (
181179 tox_ini_path : Path ,
182180 workflow_directory_path : Path ,
183- ) -> list :
181+ ) -> int :
184182 return _generate_workflow (
185183 get_lint_job_datas (get_tox_envs (tox_ini_path )),
186184 "lint" ,
@@ -191,7 +189,7 @@ def generate_lint_workflow(
191189def generate_misc_workflow (
192190 tox_ini_path : Path ,
193191 workflow_directory_path : Path ,
194- ) -> list :
192+ ) -> int :
195193 return _generate_workflow (
196194 get_misc_job_datas (get_tox_envs (tox_ini_path )),
197195 "misc" ,
@@ -200,9 +198,9 @@ def generate_misc_workflow(
200198
201199
202200def generate_ci_workflow (
203- test_file_numbers : list ,
204- lint_file_numbers : list ,
205- misc_file_numbers : list ,
201+ test_count : int ,
202+ lint_count : int ,
203+ misc_count : int ,
206204 output_dir : Path ,
207205) -> None :
208206 """Generate the parent CI orchestrator workflow."""
@@ -215,9 +213,9 @@ def generate_ci_workflow(
215213 )
216214 .get_template ("ci.yml.j2" )
217215 .render (
218- test_file_numbers = test_file_numbers ,
219- lint_file_numbers = lint_file_numbers ,
220- misc_file_numbers = misc_file_numbers ,
216+ test_count = test_count ,
217+ lint_count = lint_count ,
218+ misc_count = misc_count ,
221219 )
222220 )
223221 ci_yml_file .write ("\n " )
@@ -226,11 +224,9 @@ def generate_ci_workflow(
226224if __name__ == "__main__" :
227225 tox_ini_path = Path (__file__ ).parent .parent .parent .joinpath ("tox.ini" )
228226 output_dir = Path (__file__ ).parent
229- test_file_numbers = generate_test_workflow (
227+ test_count = generate_test_workflow (
230228 tox_ini_path , output_dir , ["ubuntu-latest" , "windows-latest" ]
231229 )
232- lint_file_numbers = generate_lint_workflow (tox_ini_path , output_dir )
233- misc_file_numbers = generate_misc_workflow (tox_ini_path , output_dir )
234- generate_ci_workflow (
235- test_file_numbers , lint_file_numbers , misc_file_numbers , output_dir
236- )
230+ lint_count = generate_lint_workflow (tox_ini_path , output_dir )
231+ misc_count = generate_misc_workflow (tox_ini_path , output_dir )
232+ generate_ci_workflow (test_count , lint_count , misc_count , output_dir )
0 commit comments