Add typehints

2023-07-14 14:43:47 +01:00
parent c57ed6edcd
commit e522e56ff1
6 changed files with 8 additions and 8 deletions
--- a/src/primaite/config/training_config.py
+++ b/src/primaite/config/training_config.py
@@ -216,7 +216,7 @@ class TrainingConfig:
                config_dict[key] = value[config_dict[key]]
        return TrainingConfig(**config_dict)

-    def to_dict(self, json_serializable: bool = True):
+    def to_dict(self, json_serializable: bool = True) -> Dict:
        """
        Serialise the ``TrainingConfig`` as dict.