Modify docker related commands (#54)

* Add new ros package installation(orocos, hrl-kdl)

* Add source bashrc

* update train script

* Add custom package ros

* Change delegating env_name logic

* Add reacher env when training selection

* Change clone branch to master
This commit is contained in:
Whi Kwon authored and Seungjae Ryan Lee committed 2019-05-21 13:26:52 +09:00
1 parent 5ff59a280e
commit 4b56ae781b
3 files changed
+18 -5

No files matched your search

+4 -1
View File
@@ -44,7 +44,10 @@ class AbstractAgent(object):
self.args.max_episode_steps = env._max_episode_steps
# for logging
self.env_name = str(self.env.env).split("<")[1].replace(">>", "")
if hasattr(env, "env_name"):
self.env_name = env.env_name
else:
self.env_name = self.env.unwrapped.spec.id
self.sha = (
subprocess.check_output(["git", "rev-parse", "--short", "HEAD"])[:-1]
.decode("ascii")