ponder-lab / Hybridize-Functions-Refactoring / 26980133543

78%

push

github

Gate `input_signature` emission on dtype-constant reachability (#585) (#587)

On the `from tensorflow import ...` named-import path, `getImportContext`
marked `tensorSpecReachable` from the presence of `TensorSpec` alone, but the
emitted `dtype=<name>` constant (e.g. `float32`) must also be importable
unqualified. A file importing `function` and `TensorSpec` but not the inferred
dtype emitted `@function(input_signature=[TensorSpec(shape=(), dtype=float32)])`,
a `NameError`-raising decorator.

Thread the named TensorFlow imports into `ImportContext` and expose the
signature's referenced dtype names via `InputSignature.requiredDTypeNames`. The
emission gate now skips when any required dtype constant is out of scope; the
named-import shape has no module prefix to qualify with. Qualified and wildcard
shapes set `allNamesReachable`, so they are unaffected.

Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

13 of 14 new or added lines in 2 files covered. (92.86%)

1303 of 1705 relevant lines covered (76.42%)

0.76 hits per line