Added import for Path

danijar · Feb 8, 2024 · ae98625 · ae98625
1 parent 67fb4b4
commit ae98625
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dreamerv3/agent.py b/dreamerv3/agent.py
@@ -2,6 +2,7 @@
 import jax
 import jax.numpy as jnp
 import ruamel.yaml as yaml
+from pathlib import Path
 tree_map = jax.tree_util.tree_map
 sg = lambda x: tree_map(jax.lax.stop_gradient, x)