Fix constraint solve!(::T)::T

tkf · Jun 19, 2018 · 0080239 · 0080239
1 parent e88112e
commit 0080239
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/continuations/interface.jl b/src/continuations/interface.jl
@@ -136,5 +136,5 @@ function solve!(wrapper::AbstractContinuationSolver)
         end
     end
 
-    return solver
+    return wrapper
 end
diff --git a/src/plotting.jl b/src/plotting.jl
@@ -1,6 +1,6 @@
 using RecipesBase
 
-using .Continuations: ContinuationSolution, sweeps_as_vectors
+using .Continuations: as, ContinuationSolution, sweeps_as_vectors
 
 @recipe function f(sol::ContinuationSolution; vars = (0, 1))
     ix, iy = vars
@@ -10,3 +10,10 @@ using .Continuations: ContinuationSolution, sweeps_as_vectors
     delete!(plotattributes, :vars)
     (xs, ys)
 end
+
+using .Codim1: Codim1Solution
+
+@recipe function f(sol::Codim1Solution; vars = (0, 1))
+    super = as(sol, ContinuationSolution)
+    super
+end