Fix super() compatibility

This commit fixes the super() compatibility between Python 2 and 3 versions. The problem was resolved by passing the class name and an instance of self to the super() as arguments. TODO: There should be an instantiation checker for the search methods. When calling assertions() from base., please use super(ClassName, self).assertions() Author: ljvmiranda921 Email: ljvmiranda921@gmail.com
ljvmiranda921 · Sep 15, 2017 · 39d8358 · 39d8358
1 parent 7444c09
commit 39d8358
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyswarms/utils/search/grid_search.py b/pyswarms/utils/search/grid_search.py
@@ -68,7 +68,7 @@ def __init__(self, optimizer, n_particles, dimensions, options,
         """Initializes the paramsearch."""
 
         # Assign attributes
-        super().__init__(optimizer, n_particles, dimensions, options,
+        super(GridSearch, self).__init__(optimizer, n_particles, dimensions, options,
                 objective_func, iters, bounds=bounds,
                 velocity_clamp=velocity_clamp)
 

diff --git a/pyswarms/utils/search/random_search.py b/pyswarms/utils/search/random_search.py
@@ -79,7 +79,7 @@ def __init__(self, optimizer, n_particles, dimensions, options,
         """
 
         # Assign attributes
-        super().__init__(optimizer, n_particles, dimensions, options,
+        super(RandomSearch, self).__init__(optimizer, n_particles, dimensions, options,
                 objective_func, iters, bounds=bounds,
                 velocity_clamp=velocity_clamp)
         self.n_selection_iters = n_selection_iters