Skip to content

Commit b2bc95c

Browse files
committed
Python2 fix
1 parent f55ddc4 commit b2bc95c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/ppo/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def __init__(self, lr, brain, h_size, epsilon, max_step, normalize, num_layers):
204204
:param brain: State-space size
205205
:param h_size: Hidden layer size
206206
"""
207-
super().__init__()
207+
super(ContinuousControlModel, self).__init__()
208208
s_size = brain.state_space_size
209209
a_size = brain.action_space_size
210210

@@ -269,7 +269,7 @@ def __init__(self, lr, brain, h_size, epsilon, beta, max_step, normalize, num_la
269269
:param brain: State-space size
270270
:param h_size: Hidden layer size
271271
"""
272-
super().__init__()
272+
super(DiscreteControlModel, self).__init__()
273273
self.create_global_steps()
274274
self.create_reward_encoder()
275275
self.normalize = normalize

0 commit comments

Comments
 (0)