Browse code

fix bug with saving of the BalancingLearner

Bas Nijholt authored on 27/11/2018 13:59:26
Showing 1 changed files
... ...
@@ -350,7 +350,7 @@ class BalancingLearner(BaseLearner):
350 350
         >>> learner.save(combo_fname)  # use 'load' in the same way
351 351
         """
352 352
         if isinstance(fname, Iterable):
353
-            for l, _fname in zip(fname, self.learners):
353
+            for l, _fname in zip(self.learners, fname):
354 354
                 l.save(_fname, compress=compress)
355 355
         else:
356 356
             for l in self.learners:
... ...
@@ -374,7 +374,7 @@ class BalancingLearner(BaseLearner):
374 374
         See the example in the `BalancingLearner.save` doc-string.
375 375
         """
376 376
         if isinstance(fname, Iterable):
377
-            for l, _fname in zip(fname, self.learners):
377
+            for l, _fname in zip(self.learners, fname):
378 378
                 l.load(_fname, compress=compress)
379 379
         else:
380 380
             for l in self.learners: