fix: features transform in Trainer._save_model internal method#573
fix: features transform in Trainer._save_model internal method#573
Trainer._save_model internal method#573Conversation
DaniBodor
left a comment
There was a problem hiding this comment.
Nice catch, once again.
If ever there is some extra time on this project (not likely), it would be nice to refactor the entire tests folder. Many of these modules are getting really long and confusing and contain lots of duplicate code. It's hard to keep track of what is going on. A nice cleanup could make it really easy to quickly assess what each test does and how.
Indeed, it would be very nice. We also have already issues for that. Let's see what budgets we get ;) |
transformfunction for handling both cases. The bug was causing the failure of thetraining.ipynbnotebook.features_transformattribute in theTrainer, and in case we use a pre-trained model. The testing on the trainer side for features_transform was missing.As soon as this is approved, I will release a new patch version for deeprank2.