added unregister function for edges

plasmo-dev · Aug 21, 2024 · e5af7d5 · e5af7d5
1 parent 0e6d239
commit e5af7d5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/optiedge.jl b/src/optiedge.jl
@@ -94,6 +94,10 @@ function JuMP.delete(graph::OptiGraph, cref::ConstraintRef)
     #TODO: Probably need to delete the edge altogether if it is the only constraint on the edge
 end
 
+function JuMP.unregister(graph::OptiGraph, key::Symbol)
+    delete!(object_dictionary(graph), key)
+end
+
 ### Edge Constraints
 
 # NOTE: could use one method for node and edge