[lslusarczyk]

inspect() now measures module depth in executed modules, and get_modules() covers the whole model at any depth.

Depth used to count object path segments.
nn.ModuleList, plain containers, and modules that never ran each added a level.
Two depths could return the same modules, and a deeper one could return none.
get_modules() dropped executed modules shallower than min_depth, so those parts of the model were never tuned.
It also reset forward_called, so it worked only once per inspection.

Depth is now the number of executed modules above a module.
get_modules(depth) returns every executed module at that depth, plus shallower executed modules without executed children.
It leaves the inspection unchanged, so one inspection serves several depths.
max_depth ignores modules deeper than that depth.

This is a breaking change for min_depth.
A diffusers pipeline that used min_depth=1 to reach its components now uses min_depth=0, so benchmark_hybrid.py drops it.
The Parakeet and E5Large examples keep their values: their modules map to the same depths.

[Copilot]

Warning

Copilot couldn't run its full agentic review because it didn't start before the timeout. Make sure your repository has a runner available, or add a copilot-code-review.yml file specifying one with the runs-on attribute. See the docs for more details.

Copilot review overview

4 open issues

• dataclasses.replace() only works on dataclass instances. In this diff, InspectContext is… · New
• The module paths in this table omit the leading dot that the library’s object-path examples use… · New
• depth is computed recursively on every access (and calls executed_parent, which also walks up… · New
• Correct the typo tunning to tuning in this comment. · New

What changed in this PR

Updates inspecting depth semantics to be based on executed-module ancestry (not object-path segments), and changes the public API from min_depth/max_depth to a single depth cut.

Changes:

• Redefines module “depth” as the count of executed ancestors and updates get_modules() to cut the executed graph consistently.
• Reworks traversal bounding to use object-path depth derived from the requested cut depth, removing max_depth.
• Updates docs, examples, and unit tests to the new depth API and semantics.

File	Description
torch_tweak/​torch/​inspecting/​module_inspector.py	Implements new cut logic and replaces recursion-depth limit with path-depth limit derived from depth.
torch_tweak/​torch/​inspecting/​module_info.py	Introduces computed executed_parent and depth based on executed ancestors.
torch_tweak/​torch/​inspecting/​inspecting.py	Switches API to depth and updates docstring/instantiation.
tests/​unit/​torch/​inspecting/​test_wrapping.py	Updates a wrapping test to use the new depth value.
tests/​unit/​torch/​inspecting/​test_module_inspector.py	Removes old max-depth test and adds new depth semantics tests (executed-only depth, stable get_modules, etc.).
examples/​StableDiffusion/​stable_diffusion/​benchmark_hybrid.py	Drops min_depth use to align with new semantics (default depth=0).
examples/​ParakeetRNNT/​parakeet_rnnt/​tune.py	Replaces min_depth with depth.
examples/​ParakeetCTC/​parakeet_ctc/​tune.py	Replaces min_depth with depth.
examples/​E5Large/​e5large/​tune.py	Replaces min_depth with depth and updates the comment accordingly.
docs/​guides/​inspect.md	Documents the new executed-module depth semantics and the depth parameter.
ChangeLog.md	Notes breaking change and describes new depth behavior/API.

_{Review effort: Lite}

[lukaszstolarczuk]

dco is failing

[lslusarczyk]

dco is failing

rebased and fixed