Comparator fix for weakref

[aseembits93]

Some extra tests to ensure more torch datatypes are supported but no changes in the comparator

[claude]

PR Review Summary

Prek Checks

✅ All checks pass (ruff check and ruff format both passed).

Mypy

⚠️ 426 mypy errors found in the two changed files, but these are all pre-existing (missing return type annotations in test file, no-any-return errors in comparator.py). No new type errors introduced by this PR.

Code Review

✅ No critical issues found in the latest changes.

Changes reviewed:

• codeflash/verification/comparator.py: Adds weakref import and handler for weakref.ref objects (lines 175-184). Adds type annotation for superset_obj parameter.
• tests/test_comparator.py: Adds comprehensive tests for weakref handling (3 test functions) and torch.nn module comparisons (16 test functions).

Analysis:

• The weakref handler correctly dereferences both refs and compares referents recursively
• Dead ref handling (both dead = equal, one dead = not equal) is correct
• The type guard at line 127-132 ensures new is also a weakref.ref when orig is one
• Potential infinite recursion from circular weakrefs is caught by the existing RecursionError handler at line 528
• torch.nn tests properly use pytest.skip() when torch is not available

Test Coverage

File	Stmts (main)	Stmts (PR)	Coverage (main)	Coverage (PR)	Change
codeflash/verification/comparator.py	332	341	61%	62%	+1% ✅

• New weakref lines (175-184): ✅ Fully covered by tests
• Overall: Coverage improved slightly (+1%), no regressions
• Skipped tests: 42 skipped (vs 19 on main) — the new torch.nn tests are skipped when torch is not installed in CI, which is expected

---

Last updated: 2026-02-11