import custom_types in uncertainty_metrics

adaptive-intelligent-robotics · Sep 10, 2024 · 7450914 · 7450914
1 parent 433e6be
commit 7450914
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qdax/utils/uncertainty_metrics.py b/qdax/utils/uncertainty_metrics.py
@@ -5,7 +5,7 @@
 import jax.numpy as jnp
 
 from qdax.core.containers.mapelites_repertoire import MapElitesRepertoire
-from qdax.types import Descriptor, ExtraScores, Fitness, Genotype, RNGKey
+from qdax.custom_types import Descriptor, ExtraScores, Fitness, Genotype, RNGKey
 from qdax.utils.sampling import (
     dummy_extra_scores_extractor,
     median,