search

harshraj22 / rl_lab

Contains Solutions of Lab assignments of Reinforcement Learning lab
0 stars 0 forks source link

shorten code #16

Closed harshraj22 closed 2 years ago

harshraj22 commented 2 years ago

https://github.com/harshraj22/rl_lab/blob/a97e10df1a3a9588337119d814d2d6364107bd3f/lab3/data_loader/environments.py#L102

Assign reward and done in 1 line. Shorten the code