def get_sparsity_in(self, i): n = nlpsol_out(i) if n == "f": return Sparsity.scalar() elif n in ("x", "lam_x"): return Sparsity.dense(self.nx) elif n in ("g", "lam_g"): return Sparsity.dense(self.ng) else: return Sparsity(0, 0)
def get_sparsity_in(self, i: int) -> tuple: """ Get the sparsity of a specific variable Parameters ---------- i: int The index of the variable Returns ------- The sparsity of the variable """ n = nlpsol_out(i) if n == "f": return Sparsity.scalar() elif n in ("x", "lam_x"): return Sparsity.dense(self.nx) elif n in ("g", "lam_g"): return Sparsity.dense(self.ng) else: return Sparsity(0, 0)