some plotting ax returns

Author: Datseris

Project.toml

@@ -1,7 +1,7 @@
 name = "DynamicalBilliards"
 uuid = "4986ee89-4ee5-5cef-b6b8-e49ba721d7a5"
 repo = "https://github.com/JuliaDynamics/DynamicalBilliards.jl.git"
-version = "3.11.2"
+version = "3.11.3"
 
 [deps]
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"

src/plotting/billiards.jl

@@ -36,7 +36,7 @@ Works for both periodic and normal billiards.
 function plot(bd::Billiard) end
 
 function plot(bd::Billiard{T};
-    ax = (PyPlot.figure(); PyPlot.gca())) where {T}
+    ax = (PyPlot.figure(); PyPlot.gca())) where {T <: AbstractFloat}
     PyPlot.sca(ax)
     for obst in bd; plot(obst); end
     xmin, ymin, xmax, ymax = cellsize(bd)
@@ -48,7 +48,7 @@ function plot(bd::Billiard{T};
     if !isinf(ymin) && !isinf(ymax)
         PyPlot.ylim(ymin - 0.1dy, ymax + 0.1dy)
     end
-    return nothing
+    return ax
 end
 
 function plot(bd::Billiard, xmin, ymin, xmax, ymax;
@@ -72,7 +72,7 @@ function plot(bd::Billiard, xmin, ymin, xmax, ymax;
 
     PyPlot.xlim(xmin, xmax)
     PyPlot.ylim(ymin, ymax)
-    return nothing
+    return ax
 end
 
 function plot(bd, xt::AbstractVector, yt::AbstractVector;
@@ -114,7 +114,7 @@ function plot(bd, xt::AbstractVector, yt::AbstractVector;
             PyPlot.gca().set_aspect("equal")
         end
     end
-    return nothing
+    return ax
 end
 
 function plot_periodic_rectangle(bd, xmin, ymin, xmax, ymax)