fix: add fixes for pymoo test

Author: harish-panneerselvam

src/t3co/cli/sweep.py

@@ -24,7 +24,10 @@
 from t3co.utils.print_class_objects import get_path_object
 
 try:
-    from pymoo.core.problem import StarmapParallelization
+    try:
+        from pymoo.parallelization.starmap import StarmapParallelization
+    except ImportError:
+        from pymoo.core.problem import StarmapParallelization
     from pymoo.optimize import minimize
     from t3co.optimize.optimization import (
         VehicleDesignOpt,
@@ -36,8 +39,6 @@
 
 except ImportError:
     optimization_installed = False
-except AttributeError:
-    optimization_installed = False
 
 
 def _normalize_selections_arg(selections) -> list | None:

src/t3co/optimize/optimization.py

@@ -6,7 +6,12 @@
 from pymoo.algorithms.soo.nonconvex.nelder import NelderMead
 from pymoo.algorithms.soo.nonconvex.pattern import PatternSearch
 from pymoo.algorithms.soo.nonconvex.pso import PSO
-from pymoo.core.problem import ElementwiseProblem, StarmapParallelization
+from pymoo.core.problem import ElementwiseProblem
+
+try:
+    from pymoo.parallelization.starmap import StarmapParallelization
+except ImportError:
+    from pymoo.core.problem import StarmapParallelization
 from pymoo.optimize import minimize
 from pymoo.termination.default import DefaultMultiObjectiveTermination as MODT