Skip to content

Commit 49de038

Browse files
committed
fix condition
1 parent 4d0dd91 commit 49de038

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

src/gradient_free_optimizers/_estimators/_gaussian_process_regressor.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -121,11 +121,10 @@ def fit(self, X, y):
121121
def _add_diagonal(self, K, value):
122122
"""Add value to diagonal of matrix K."""
123123
n = len(K)
124-
# Convert to list of lists for manipulation
125-
if hasattr(K, "_data"):
126-
result = [list(row) for row in K._data]
127-
elif hasattr(K, "tolist"):
124+
if hasattr(K, "tolist"):
128125
result = K.tolist()
126+
elif hasattr(K, "_data"):
127+
result = [list(row) for row in K]
129128
else:
130129
result = [list(row) for row in K]
131130

@@ -276,7 +275,9 @@ def _matmul_mv(self, A, v):
276275

277276
def _sum_squared_cols(self, v):
278277
"""Sum of squared values along columns (axis=0)."""
279-
if hasattr(v, "_data"):
278+
if hasattr(v, "tolist"):
279+
data = v.tolist()
280+
elif hasattr(v, "_data"):
280281
data = v._data
281282
else:
282283
data = list(v)

0 commit comments

Comments
 (0)