Merge pull request #765 from JuliaReach/schillic/affine_map

Make affine_map consistent
JuliaReach · Jan 25, 2024 · 2d7e7c1 · 2d7e7c1
2 parents 296c76e + a17f395
commit 2d7e7c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/Flowpipes/Flowpipe.jl b/src/Flowpipes/Flowpipe.jl
@@ -270,7 +270,7 @@ function LazySets.linear_map(M, fp::Flowpipe)
     return Flowpipe(out, fp.ext)
 end
 
-function LazySets.affine_map(M, b, fp::Flowpipe)
+function LazySets.affine_map(M, fp::Flowpipe, b)
     out = [reconstruct(R, affine_map(M, set(R), b)) for R in fp]
     return Flowpipe(out, fp.ext)
 end

diff --git a/src/Flowpipes/MixedFlowpipe.jl b/src/Flowpipes/MixedFlowpipe.jl
@@ -82,7 +82,7 @@ function LazySets.linear_map(M, fp::MixedFlowpipe)
     return MixedFlowpipe(out, fp.ext)
 end
 
-function LazySets.affine_map(M, b, fp::MixedFlowpipe)
-    out = [affine_map(M, b, F) for F in fp]
+function LazySets.affine_map(M, fp::MixedFlowpipe, b)
+    out = [affine_map(M, F, b) for F in fp]
     return MixedFlowpipe(out, fp.ext)
 end