File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments