Skip to content

Update test imports for v3 model architecture#7

Open
ShadNygren wants to merge 1 commit into
Kwaai-AI-Lab:mainfrom
ShadNygren:test/update-imports-for-v3-model
Open

Update test imports for v3 model architecture#7
ShadNygren wants to merge 1 commit into
Kwaai-AI-Lab:mainfrom
ShadNygren:test/update-imports-for-v3-model

Conversation

@ShadNygren
Copy link
Copy Markdown
Contributor

Summary

  • Update test_ibm_pd_ad.py to import from IBM_PD_AD_v3 instead of IBM_PD_AD
  • Update age band tests for v3 model bands: (65, 79) and (80, None)
  • Update test_cvd_figures.py to import from CVD_Study/ directory
  • Update test_psa_workflow.py to verify v3 PSA architecture (replaces obsolete psa_with_timeseries tests)

Test plan

  • All 139 tests pass
  • Flake8 linting passes
  • Verified imports work with new file locations

🤖 Generated with Claude Code

- Update test_ibm_pd_ad.py to import from IBM_PD_AD_v3 instead of IBM_PD_AD
- Update age band tests for v3 model bands: (65, 79) and (80, None)
- Update test_cvd_figures.py to import from CVD_Study/ directory
- Update test_psa_workflow.py to verify v3 PSA architecture
- Remove obsolete psa_with_timeseries dependency tests
- Add tests verifying integrated PSA in IBM_PD_AD_v3

All 139 tests pass with the updated imports.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant