Fix bug get* functions

thieu1995 · Apr 25, 2024 · 7ae62ff · 7ae62ff
1 parent 67af338
commit 7ae62ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/opfunu/__init__.py b/opfunu/__init__.py
@@ -99,7 +99,7 @@ def get_functions_based_ndim(ndim=None):
     """
     functions = [cls for classname, cls in ALL_DATABASE if classname not in EXCLUDES]
     if type(ndim) is int and ndim > 1:
-        return list(filter(lambda f: ndim in f().dim_supported, functions))
+        return list(filter(lambda f: (f().dim_default == ndim or f().dim_changeable == True), functions))
     return functions
 
 

diff --git a/opfunu/cec_based/cec.py b/opfunu/cec_based/cec.py
@@ -68,7 +68,7 @@ def __init__(self):
         self.dim_changeable = True
         self.dim_default = 30
         self.dim_max = 100
-        self.dim_supported = []
+        self.dim_supported = None
         self.f_global = None
         self.x_global = None
         self.n_fe = 0