polish(rjy): fix format

Author: nighood

ding/rl_utils/adder.py

@@ -130,7 +130,7 @@ def get_nstep_return_data(
             if cum_reward:
                 data[i]['reward'] = sum([data[i + j]['reward'] * (gamma ** j) for j in range(nstep)])
             else:
-                data[i]['reward'] = torch.stack([data[i + j]['reward'] for j in range(nstep)], dim = -1)
+                data[i]['reward'] = torch.stack([data[i + j]['reward'] for j in range(nstep)], dim=-1)
             data[i]['done'] = data[i + nstep - 1]['done']
             if correct_terminate_gamma:
                 data[i]['value_gamma'] = gamma ** nstep
@@ -143,7 +143,7 @@ def get_nstep_return_data(
                 data[i]['reward'] = torch.stack(
                     [data[i + j]['reward']
                      for j in range(len(data) - i)] + [fake_reward for _ in range(nstep - (len(data) - i))],
-                    dim = -1
+                    dim=-1
                 )
             # try:
             #     assert len(data[i]['reward']) == 300