fix a bug in EvoX monitor

2024-08-14 15:07:29 +08:00
parent 9cf17f8d69
commit a9c0e1bd9c
1 changed files with 2 additions and 2 deletions
--- a/src/tensorneat/common/evox_adaptors/tensorneat_monitor.py
+++ b/src/tensorneat/common/evox_adaptors/tensorneat_monitor.py
@@ -84,7 +84,7 @@ class TensorNEATMonitor(Monitor):

        if self.is_save:
            # save best
-            best_genome = jax.device_get((pop[0][max_idx], pop[1][max_idx]))
+            best_genome = jax.device_get(self.best_genome)
            file_name = os.path.join(
                self.genome_dir, f"{generation}.npz"
            )