search

DeepRank / deeprank2

An open-source deep learning framework for data mining of protein-protein interfaces or single-residue variants.
https://deeprank2.readthedocs.io/en/latest/?badge=latest
Apache License 2.0
38 stars 10 forks source link

Fix `features_transform` AttributeError in `Trainer._save_model` method #570

Closed gcroci2 closed 8 months ago

gcroci2 commented 8 months ago

When a Trainer instance calls the _save_model method and self.features_transform has transform keys with not-None values, it raises: AttributeError: 'NoneType' object has no attribute 'group'. We didn't notice because even if we are widely testing features_transform usage within the dataset.py quite broadly, we're not doing that with the trainer.py-related usage. We should: