Refine test_step formatting and trainer configuration

- Update test_step method signature formatting in models.py for better readability
- Simplify on_test_start by removing unnecessary test_config logging
- Add trainer.test call in train_hnet.py to execute testing after training

hungarian_net/models.py

@@ -121,7 +121,9 @@ def validation_step(self, batch, batch_idx) -> Dict[str, torch.Tensor]:
 
         return outputs
 
-    def test_step(self, batch: Dict[str, torch.Tensor], batch_idx) -> Dict[str, torch.Tensor]:
+    def test_step(
+        self, batch: Dict[str, torch.Tensor], batch_idx
+    ) -> Dict[str, torch.Tensor]:
         """
         Lightning test step.
 
@@ -157,10 +159,7 @@ def on_test_start(self) -> None:
         # 1. Reset test metrics
         self.f1.reset()
 
-        # 2. Log test configurations
-        self.logger.log_dict({"test_config": self.hparams})
-
-        # 3. Disable certain training-specific settings
+        # 2. Disable certain training-specific settings
         self.model.eval()
 
     def on_train_batch_end(

hungarian_net/train_hnet.py

@@ -163,6 +163,8 @@ def main(
 
     trainer.fit(lightning_module, datamodule=lightning_datamodule)
 
+    trainer.test(ckpt_path="best", datamodule=lightning_datamodule)
+
 
 def set_seed(seed=42):
     L.seed_everything(seed, workers=True)