From 9e0f8c579696fb44dd5ce02bfc9f09b4f6c6467a Mon Sep 17 00:00:00 2001
From: Alinson S Xavier <axavier@anl.gov>
Date: Thu, 27 Jul 2023 10:38:36 -0500
Subject: [PATCH] solve: Allow custom graph

---
 src/model/solve.jl | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/model/solve.jl b/src/model/solve.jl
index 1758e08..6d03dda 100644
--- a/src/model/solve.jl
+++ b/src/model/solve.jl
@@ -32,6 +32,7 @@ function solve(
     output = nothing,
     marginal_costs = true,
     return_model = false,
+    graph = nothing,
 )
 
     if lp_optimizer == nothing
@@ -51,7 +52,9 @@ function solve(
 
 
     @info "Building graph..."
-    graph = RELOG.build_graph(instance)
+    if graph === nothing
+        graph = RELOG.build_graph(instance)
+    end
     _print_graph_stats(instance, graph)
 
     @info "Building optimization model..."