Address code review feedback: move imports to top, improve comments

Co-authored-by: functionstackx <47992694+functionstackx@users.noreply.github.com>

Author: Copilot

.github/workflows/example-batched-matrix.yml

@@ -47,6 +47,9 @@ jobs:
     strategy:
       matrix:
         # Generate array [0, 1, 2, ..., batch-count-1]
+        # This expression creates a comma-separated list of indices and wraps it in an array
+        # Example: if batch-count=3, this generates: [0, 1, 2]
+        # Note: This is a workaround for GitHub Actions' lack of native range array generation
         batch-index: ${{ fromJson(format('[{0}]', join(range(0, fromJson(needs.get-batch-count.outputs.batch-count)), ','))) }}
     outputs:
       # Each batch gets its own output

utils/matrix-logic/generate_sweep_configs.py

@@ -1,6 +1,7 @@
 import json
 import yaml
 import argparse
+import math
 from pydantic import BaseModel, Field, ValidationError, ConfigDict
 from typing import List
 
@@ -980,7 +981,6 @@ def main():
     validate_matrix_output(matrix_values)
 
     # Calculate number of batches
-    import math
     total_batches = math.ceil(len(matrix_values) / args.max_batch_size)
 
     # If only batch count requested, output it and exit

utils/matrix-logic/test_generate_sweep_configs.py

@@ -1,5 +1,6 @@
 import pytest
 import yaml
+import math
 from unittest.mock import patch
 from generate_sweep_configs import (
     validate_master_configs_structure,
@@ -1669,7 +1670,6 @@ def test_main_with_get_batch_count(temp_config_files):
                            '--get-batch-count']):
         batch_count = main()
 
-    import math
     expected_batches = math.ceil(len(all_results) / 2)
     assert batch_count == expected_batches
 
@@ -1699,7 +1699,6 @@ def test_main_batch_covers_all_configs(temp_config_files):
 
     # Get results in batches of 2
     batch_size = 2
-    import math
     num_batches = math.ceil(len(all_results) / batch_size)
 
     batched_results = []