Skip to content

Commit 3a0e70d

Browse files
committed
Typing fix (L434).
1 parent ff61917 commit 3a0e70d

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

linopy/dual.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def _lookup(
8686

8787
coord_values = [labels.coords[d].values for d in labels.dims]
8888

89-
# Choosing np.ndindex over np.argwhere or da.to_series for memory efficiency on large ND arrays
89+
# Choosing np.ndindex over np.argwhere or da.to_series for memory efficiency on large n-dimensional arrays
9090
for idx in np.ndindex(vals.shape):
9191
label = int(vals[idx])
9292
if label == -1:
@@ -430,7 +430,9 @@ def _add_dual_feasibility_constraints(
430430
)
431431

432432
def __rule(m: Model, *coord_vals: Any) -> LinearExpression | None:
433-
coord_dict = dict(zip(var.labels.dims, coord_vals))
433+
coord_dict = {
434+
str(dim): val for dim, val in zip(var.labels.dims, coord_vals)
435+
}
434436
flat = var.labels.sel(**coord_dict).item()
435437
if flat == -1 or flat not in (term_dict := dual_feas_terms[var_name]):
436438
return None

0 commit comments

Comments
 (0)