Skip to content

Add PDGrapher forward and inverse perturbation models#239

Draft
fctb12 wants to merge 1 commit intomainfrom
codex/integrate-pdgrapher-model-into-state
Draft

Add PDGrapher forward and inverse perturbation models#239
fctb12 wants to merge 1 commit intomainfrom
codex/integrate-pdgrapher-model-into-state

Conversation

@fctb12
Copy link
Collaborator

@fctb12 fctb12 commented Oct 28, 2025

Summary

  • add a PDGrapherLightningModule with forward and inverse graph-based perturbation heads and shared configuration helpers
  • provide Hydra presets for both PDGrapher forward and inverse training runs
  • update tx inference to instantiate arbitrary lightning modules via get_lightning_module and support PDGrapher inverse outputs

Testing

  • pytest tests/test_callbacks.py (fails: ModuleNotFoundError: No module named 'state')

https://chatgpt.com/codex/tasks/task_e_6901147f26888325a47718776ce9c99e

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

1 participant