diff --git a/torchbenchmark/models/soft_actor_critic/sac.py b/torchbenchmark/models/soft_actor_critic/sac.py index dd590a8d0..c67da4fca 100644 --- a/torchbenchmark/models/soft_actor_critic/sac.py +++ b/torchbenchmark/models/soft_actor_critic/sac.py @@ -3,7 +3,7 @@ import numpy as np import torch -from . import envs, nets, replay, utils +from . import envs, nets, replay, sac_utils class SACAgent: