Skip to content

Conversation

@CheeseLee888
Copy link
Collaborator

Refactor Lightning modules to remove the RegressionBase inheritance.

The RegressionBase class had become a “bloated” base, mixing concerns that were not shared by all Lightning modules.

Each Lightning module that previously depended on RegressionBase is now a standalone LightningModule with its own clearly defined hooks (training/validation/test/predict, logging, etc.).

Shared utilities are kept as small helper functions instead of an implicit base class dependency.

This makes the modules easier to read, extend, and test in isolation, and reduces the coupling between future changes in regression logic and the Lightning modules themselves.

@cnellington cnellington merged commit 0e0478f into AdaptInfer:main Dec 11, 2025
3 checks passed
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.

2 participants