simplify scope tracking in burn-import

[skewballfox]

Pull Request Template

Checklist

• Confirmed that run-checks all script has been executed.
• Made sure the book is up to date with changes in this PR.

Changes

Something from #2008 worth keeping around. Scopes was storing variable references as a vector, but then only storing one Tensor Variable in each entry. I changed scopes to be a single TensorVariable per entry. Makes it a little easier to understand at a glance.

the other change was to check the graph outputs and the graph inputs in the same iteration when scope checking. Since the graph is a DAG, that's guaranteed not to break.

Testing

I ran test -p burn-import, test -p onnx-test, and typos crates/burn-import. This should cover all the relevant test from run checks all

[codecov]

Codecov Report

Attention: Patch coverage is 88.23529% with 2 lines in your changes missing coverage. Please review.

Project coverage is 85.56%. Comparing base (79cd3d5) to head (1890d3a).
Report is 42 commits behind head on main.

Files with missing lines	Patch %	Lines
crates/burn-import/src/burn/scope.rs	83.33%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2207      +/-   ##
==========================================
- Coverage   85.93%   85.56%   -0.38%     
==========================================
  Files         695      707      +12     
  Lines       89032    89015      -17     
==========================================
- Hits        76514    76169     -345     
- Misses      12518    12846     +328     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[laggui]

LGTM! Just a small change needed (especially since it'll annoy clippy on the CI).

[skewballfox]

@laggui somehow I forgot to push the changes when I made that comment. Just pushed them. It seems the current CI failure is coming from somewhere outside my changes

[laggui]

Yeah the failing CI is totally unrelated to your changes, will look into it.