Apply repository-wide Ruff formatting and minimal lint cleanup for PR #49

[Copilot]

PR #49 is blocked by the CI lint workflow because ruff format --check reports 25 non-compliant files, including the tooling files called out in the failing job logs. This update brings the branch in line with the repository’s Ruff formatting/lint expectations with mechanical changes only.

• Scope: formatting alignment with CI

  • Ran Ruff formatting across the repository paths flagged by CI.
  • Updated all 25 reported files, including:
    • transactional_emulator/tools/test_data_gen.py
    • transactional_emulator/tools/trace_gp.py
    • transactional_emulator/tools/view_mem.py
    • affected TVM testbench files under transactional_emulator/testbench/

• Minimal lint-only follow-ups required by Ruff

  • Kept follow-ups strictly to Ruff-driven fixes (unused imports/vars, quoted annotations, newline/whitespace cleanup, minor string prefix cleanup).
  • No behavioral or architectural changes were introduced.

• Representative mechanical adjustment

  # before
  _, seq, head, hlen = x.shape
  
  # after
  _, seq, head, _hlen = x.shape

Original prompt

Fix the failing CI for pull request #49 in repository AICrossSim/PLENA_Simulator.

Context:

• A failing job was identified: GitHub Actions job 77863879002 from workflow CI.
• The job log shows the failure is in the formatting check step:
  • uv run ruff format --check .
  • Log excerpt:
    • Would reformat: transactional_emulator/tools/test_data_gen.py
    • Would reformat: transactional_emulator/tools/trace_gp.py
    • Would reformat: transactional_emulator/tools/view_mem.py
    • 25 files would be reformatted, 8 files already formatted
    • Process completed with exit code 1
• The workflow definition is in .github/workflows/ci.yml and confirms the failing lint job runs:
  • checkout
  • setup uv
  • uv python install 3.12
  • uv sync --no-install-project
  • uv run ruff format --check .
  • uv run ruff check .

Goal:

• Create a pull request that fixes the CI failure by applying the repository's required Ruff formatting.
• Focus on formatting-related changes only unless a formatting command reveals additional minimal lint issues that must also be fixed for CI to pass.
• Use the existing PR branch if appropriate, or otherwise create a suitable fix branch from the repository default branch.

Files directly mentioned by the log and worth checking first:

• transactional_emulator/tools/test_data_gen.py
• transactional_emulator/tools/trace_gp.py
• transactional_emulator/tools/view_mem.py

Important guidance:

• Start by reproducing the formatting issue locally in the repo environment.
• Run Ruff formatting across the repository, not just the three named files, because the log says 25 files would be reformatted.
• After formatting, run both:
  • uv run ruff format --check .
  • uv run ruff check .
• Keep the change set as small and mechanical as possible.
• The outcome should be a PR that resolves the failing formatting job for PR Gw/integrate tilelang tvm #49.

The following is the prior conversation context from the user's chat exploration (may be truncated):

User: Please find a solution for failing job 77863879002. Use the logs, job definition, and any referenced files where the failure occurred. Keep your response focused on the solution and include code suggestions when appropriate.

Assistant: The failing job is the format check, not a test or runtime failure.

Root cause

In the CI workflow, the failing step is:

• Run ruff format check in .github/workflows/ci.yml

- name: Run ruff format check
  run: uv run ruff format --check .

The logs show:

• Would reformat: transactional_emulator/tools/test_data_gen.py
• Would reformat: transactional_emulator/tools/trace_gp.py
• Would reformat: transactional_emulator/tools/view_mem.py
• 25 files would be reformatted
• Process completed with exit code 1

So the solution is to apply the project formatter and commit the result.

Files directly referenced by the failure

• transactional_emulator/tools/test_data_gen.py
• transactional_emulator/tools/trace_gp.py
• transactional_emulator/tools/view_mem.py

Recommended fix

Run this locally at the repo root:

uv sync --no-install-project
uv run ruff format .
uv run ruff check .

Then commit the formatting changes.

Code suggestions

These files mostly fail on style/layout, not logic. Typical fixes Ruff will apply here include:

transactional_emulator/tools/test_data_gen.py

Likely reorder imports and normalize spacing around function definitions/default args.

Suggested formatted shape:

import os

import torch
from torch import nn


def generate_and_save_random_weights(
    input_dim, output_dim, filename="model_weights.pth"
):
    """
    Generates random weights for a Linear layer with the given input and output
    dimensions, and saves them in .pth format so they can be loaded with
    torch.load and used with m.load_state_dict(saved_weights).
    """
    model = nn.Linear(input_dim, output_dim)
    state_dict = model.state_dict()
    torch.save(state_dict, filename)
    print(f"Random weights saved to {filename}")


def get_weights_path(filename="model_weights.pth"):
    """
    Returns the absolute path to the weights file in the current directory.
    """
    return os.path.join(os.path.dirname(os.path.abspath(__file__)), filename)


if __name__ == "__main__":
    generate_and_save_random_weights(128, 128, get_weights_path("model_weights.pth"))

transactional_emulator/tools/trace_gp.py

This f...

This pull request was created from Copilot chat.