Skip to content

Commit 501aa2c

Browse files
authored
Merge pull request #22 from UWRobotLearning/correct-rsl-rl-version
updating rsl-rl version to >=2.3.0
2 parents 137809a + 24180fc commit 501aa2c

2 files changed

Lines changed: 3 additions & 8 deletions

File tree

source/wheeledlab_rl/setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@
1515
# NOTE: Add dependencies
1616
"psutil",
1717
"rich",
18-
"av"
18+
"av",
19+
"rsl-rl-lib>=2.3.0",
1920
]
2021

2122
# Installation operation

source/wheeledlab_rl/wheeledlab_rl/utils/modified_rsl_rl_runner.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -115,13 +115,7 @@ def learn(self, num_learning_iterations, init_at_random_ep_len=False):
115115
start = stop
116116
self.alg.compute_returns(critic_obs)
117117

118-
(
119-
mean_value_loss,
120-
mean_surrogate_loss,
121-
mean_entropy,
122-
mean_rnd_loss,
123-
mean_symmetry_loss,
124-
) = self.alg.update()
118+
loss_dict = self.alg.update()
125119
stop = time.time()
126120
learn_time = stop - start
127121
self.current_learning_iteration = it

0 commit comments

Comments
 (0)