Use ruff to modify basic.py

Author: XU-Boqing

src/evox/problems/numerical/basic.py

@@ -90,7 +90,11 @@ def _true_evaluate(self, x: torch.Tensor) -> torch.Tensor:
 
 
 def griewank_func(x: torch.Tensor) -> torch.Tensor:
-    f = 1 / 4000 * torch.sum(x**2, dim=1) - torch.prod(torch.cos(x / torch.sqrt(torch.arange(1, x.size(1) + 1, device=x.device))), dim=1) + 1
+    f = (
+        1 / 4000 * torch.sum(x**2, dim=1)
+        - torch.prod(torch.cos(x / torch.sqrt(torch.arange(1, x.size(1) + 1, device=x.device))), dim=1)
+        + 1
+    )
     return f
 
 
@@ -202,9 +206,9 @@ def ellipsoid_func(x: torch.Tensor):
 def zakharov_func(x: torch.Tensor) -> torch.Tensor:
     d = x.size(-1)
     i = torch.arange(1, d + 1, dtype=x.dtype, device=x.device)
-    sum1 = torch.sum(x ** 2, dim=-1)
+    sum1 = torch.sum(x**2, dim=-1)
     sum2 = torch.sum(0.5 * i * x, dim=-1)
-    return sum1 + sum2 ** 2 + sum2 ** 4
+    return sum1 + sum2**2 + sum2**4
 
 
 class Zakharov(ShiftAffineNumericalProblem):
@@ -220,13 +224,15 @@ def __init__(self, **kwargs):
     def _true_evaluate(self, x: torch.Tensor) -> torch.Tensor:
         return zakharov_func(x)
 
+
 def levy_func(x: torch.Tensor) -> torch.Tensor:
     w = 1 + (x - 1) / 4
     term1 = torch.sin(torch.pi * w[:, 0]) ** 2
     term2 = torch.sum((w[:, :-1] - 1) ** 2 * (1 + 10 * torch.sin(torch.pi * w[:, :-1] + 1) ** 2), dim=1)
     term3 = (w[:, -1] - 1) ** 2 * (1 + torch.sin(2 * torch.pi * w[:, -1]) ** 2)
     return term1 + term2 + term3
 
+
 class Levy(ShiftAffineNumericalProblem):
     """The Levy function whose minimum is x = [1, ..., 1]"""