chore: fix typing for untyped defs

[dodamih]

No description provided.

[codecov]

Codecov Report

Merging #449 (6bb3309) into main (c306552) will not change coverage.
Report is 7 commits behind head on main.
The diff coverage is 100.00%.

❗ Current head 6bb3309 differs from pull request most recent head ea9a241. Consider uploading reports for the commit ea9a241 to get more accurate results

@@            Coverage Diff            @@
##              main      #449   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files          120       120           
  Lines         3584      3585    +1     
=========================================
+ Hits          3584      3585    +1     

Files Changed	Coverage Δ
zetta_utils/layer/volumetric/layer_set/layer.py	100.00% <ø> (ø)
zetta_utils/parsing/ngl_state.py	100.00% <ø> (ø)
zetta_utils/tensor_ops/mask.py	100.00% <ø> (ø)
zetta_utils/builder/build.py	100.00% <100.00%> (ø)
zetta_utils/builder/built_in_registrations.py	100.00% <100.00%> (ø)
zetta_utils/geometry/vec.py	100.00% <100.00%> (ø)
zetta_utils/tensor_ops/transform.py	100.00% <100.00%> (ø)
zetta_utils/training/datasets/layer_dataset.py	100.00% <100.00%> (ø)

[dodamih]

Most of the type: ignore that are left are torchfields and docker not being typed.

[supersergiy]

Looks great!