Fix Python 2

apache · Oct 24, 2019 · ac500fc · ac500fc
1 parent 57e92d2
commit ac500fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -1515,14 +1515,14 @@ def forward(self, x):
 def test_save_load_deduplicate_with_shared_params():
     class B(mx.gluon.Block):
         def __init__(self, params=None):
-            super().__init__(params=params)
+            super(self, B).__init__(params=params)
 
             with self.name_scope():
                 self.weight = self.params.get('weight', shape=(10, 10))
 
     class C(mx.gluon.Block):
         def __init__(self, b1, b2):
-            super().__init__()
+            super(self, C).__init__()
             self.b1 = b1
             self.b2 = b2