From ec6cd8abb4c195e534467a329ad86b96ef7c90b9 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 20 Jun 2024 16:22:40 -0400 Subject: [PATCH] Rename sac utils --- torchbenchmark/models/pytorch_unet/install.py | 2 +- torchbenchmark/models/soft_actor_critic/__init__.py | 2 +- .../models/soft_actor_critic/{utils.py => sac_utils.py} | 0 3 files changed, 2 insertions(+), 2 deletions(-) rename torchbenchmark/models/soft_actor_critic/{utils.py => sac_utils.py} (100%) diff --git a/torchbenchmark/models/pytorch_unet/install.py b/torchbenchmark/models/pytorch_unet/install.py index 44bd90cc2c..d970819692 100644 --- a/torchbenchmark/models/pytorch_unet/install.py +++ b/torchbenchmark/models/pytorch_unet/install.py @@ -1,4 +1,4 @@ from utils.python_utils import pip_install_requirements if __name__ == '__main__': - pip_install_requirements() + pip_install_requirements(requirements_txt="pytorch_unet/requirements.txt") diff --git a/torchbenchmark/models/soft_actor_critic/__init__.py b/torchbenchmark/models/soft_actor_critic/__init__.py index 5008ba5c5f..1d76c2442c 100644 --- a/torchbenchmark/models/soft_actor_critic/__init__.py +++ b/torchbenchmark/models/soft_actor_critic/__init__.py @@ -12,7 +12,7 @@ from .envs import load_gym from .sac import SACAgent from .replay import PrioritizedReplayBuffer, ReplayBuffer -from .utils import hard_update, soft_update +from .sac_utils import hard_update, soft_update def learn_standard( diff --git a/torchbenchmark/models/soft_actor_critic/utils.py b/torchbenchmark/models/soft_actor_critic/sac_utils.py similarity index 100% rename from torchbenchmark/models/soft_actor_critic/utils.py rename to torchbenchmark/models/soft_actor_critic/sac_utils.py