Skip to content

Commit ae98625

Browse files
committed
Added import for Path
1 parent 67fb4b4 commit ae98625

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

dreamerv3/agent.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import jax
33
import jax.numpy as jnp
44
import ruamel.yaml as yaml
5+
from pathlib import Path
56
tree_map = jax.tree_util.tree_map
67
sg = lambda x: tree_map(jax.lax.stop_gradient, x)
78

0 commit comments

Comments
 (0)