Fix bug ndim in Benchmark class

thieu1995 · Mar 22, 2024 · 83379ff · 83379ff
1 parent 5735acb
commit 83379ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opfunu/benchmark.py b/opfunu/benchmark.py
@@ -81,7 +81,7 @@ def check_ndim_and_bounds(self, ndim=None, bounds=None, default_bounds=None):
         """
         if ndim is None:
             self._bounds = default_bounds if bounds is None else np.array(bounds).T
-            self._ndim = self._bounds.shape[1]
+            self._ndim = self._bounds.shape[0]
         else:
             if bounds is None:
                 if self.dim_changeable:
@@ -239,7 +239,7 @@ def ub(self):
         """
         return np.array([x[1] for x in self.bounds])
 
-    def create_solution(self):
+    def create_solution(self) -> np.ndarray:
         """
         Create a random solution for the current problem