fix

Author: blegat

src/graph_tools.jl

@@ -249,10 +249,11 @@ end
 Compute the sparsity pattern of the gradient of an expression (that is, a list
 of which variable indices are present).
 
-`f` is duck-typed to a `_SubexpressionStorage`-like object exposing `f.nodes`
-and `f.sizes`. For `NODE_VARIABLE_BLOCK` nodes, the block's length is read
-from `f.sizes` (via `_length`), and the block contributes that many
-consecutive variable indices starting at `nodes[k].index`.
+`f` is duck-typed (as its type is defined later) to a
+`_SubexpressionStorage`-like object exposing `f.nodes` and `f.sizes`.
+For `NODE_VARIABLE_BLOCK` nodes, the block's length is read from `f.sizes`
+(via `_length`), and the block contributes that many consecutive variable
+indices starting at `nodes[k].index`.
 """
 function _compute_gradient_sparsity!(indices::Coloring.IndexedSet, f)
     for (k, node) in enumerate(f.nodes)
@@ -263,16 +264,11 @@ function _compute_gradient_sparsity!(indices::Coloring.IndexedSet, f)
             for i in 0:(len - 1)
                 push!(indices, node.index + i)
             end
-        elseif node.type == NODE_MOI_VARIABLE
+        elseif node.type == NODE_MOI_VARIABLE || node.type == NODE_MOI_VARIABLE_BLOCK
             error(
-                "Internal error: Invalid to compute sparsity if NODE_MOI_VARIABLE " *
+                "Internal error: Invalid to compute sparsity if $(node.type) " *
                 "nodes are present.",
             )
-        elseif node.type == NODE_MOI_VARIABLE_BLOCK
-            error(
-                "Internal error: Invalid to compute sparsity if " *
-                "NODE_MOI_VARIABLE_BLOCK nodes are present.",
-            )
         end
     end
     return