Rename sac utils

pytorch · Jun 20, 2024 · ec6cd8a · ec6cd8a
1 parent 14e33ff
commit ec6cd8a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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/...nchmark/models/soft_actor_critic/utils.py → ...ark/models/soft_actor_critic/sac_utils.py b/...nchmark/models/soft_actor_critic/utils.py → ...ark/models/soft_actor_critic/sac_utils.py