Solve pre-commit errors

adaptive-intelligent-robotics · Jan 8, 2024 · dd72df7 · dd72df7
1 parent 5278ada
commit dd72df7
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/qdax/baselines/genetic_algorithm.py b/qdax/baselines/genetic_algorithm.py
@@ -122,7 +122,7 @@ def update(
             genotypes=genotypes,
             fitnesses=fitnesses,
             descriptors=None,
-            extra_scores=extra_scores | extra_info,
+            extra_scores={**extra_scores, **extra_info},
         )
 
         # update the metrics

diff --git a/qdax/core/distributed_map_elites.py b/qdax/core/distributed_map_elites.py
@@ -143,7 +143,7 @@ def update(
             genotypes=genotypes,
             fitnesses=fitnesses,
             descriptors=descriptors,
-            extra_scores=extra_scores | extra_info,
+            extra_scores={**extra_scores, **extra_info},
         )
 
         # update the metrics

diff --git a/qdax/core/emitters/multi_emitter.py b/qdax/core/emitters/multi_emitter.py
@@ -133,7 +133,7 @@ def emit(
             batch_size = jax.tree_util.tree_leaves(genotype)[0].shape[0]
             assert batch_size == emitter.batch_size
             all_offsprings.append(genotype)
-            all_extra_info = all_extra_info | extra_info
+            all_extra_info = {**all_extra_info, **extra_info}
 
         # concatenate offsprings together
         offsprings = jax.tree_util.tree_map(

diff --git a/qdax/core/emitters/qdcg_emitter.py b/qdax/core/emitters/qdcg_emitter.py
@@ -450,7 +450,7 @@ def scan_train_critics(
             length=self._config.num_critic_training_steps,
         )
 
-        return emitter_state
+        return emitter_state # type: ignore
 
     @partial(jax.jit, static_argnames=("self",))
     def _train_critics(

diff --git a/qdax/core/map_elites.py b/qdax/core/map_elites.py
@@ -143,7 +143,7 @@ def update(
             genotypes=genotypes,
             fitnesses=fitnesses,
             descriptors=descriptors,
-            extra_scores=extra_scores | extra_info,
+            extra_scores={**extra_scores, **extra_info},
         )
 
         # update the metrics