[Serving] Fix graph initialization in add model (#639)

mlrun · Jan 5, 2021 · 82b805e · 82b805e
1 parent a207597
commit 82b805e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlrun/runtimes/serving.py b/mlrun/runtimes/serving.py
@@ -251,7 +251,7 @@ def add_model(
         """
         graph = self.spec.graph
         if not graph:
-            self.set_topology()
+            graph = self.set_topology()
 
         if graph.kind != StateKinds.router:
             raise ValueError("models can only be added under router state")