From 74509145ac7119fdc692a0eb779cae1eeb0259aa Mon Sep 17 00:00:00 2001 From: Luca Grillotti Date: Tue, 10 Sep 2024 12:11:46 +0200 Subject: [PATCH] import custom_types in uncertainty_metrics --- qdax/utils/uncertainty_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qdax/utils/uncertainty_metrics.py b/qdax/utils/uncertainty_metrics.py index 2dd61c18..62a976b1 100644 --- 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,