make UnicodePlots a weakdep (for julia > 1.9)

Author: ablaom

Project.toml

@@ -12,6 +12,12 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 ScientificTypes = "321657f4-b219-11e9-178b-2701a2544e81"
 UnicodePlots = "b8865327-cd53-5732-bb35-84acbb429228"
 
+[weakdeps]
+UnicodePlots = "b8865327-cd53-5732-bb35-84acbb429228"
+
+[extensions]
+UnivariateFiniteDisplayExt = "UnicodePlots"
+
 [compat]
 CategoricalArrays = "0.9, 0.10"
 Distributions = "0.25"

src/CategoricalDistributions.jl

@@ -13,10 +13,8 @@ using OrderedCollections
 using CategoricalArrays
 import Missings
 using Random
-using UnicodePlots
 
 const Dist = Distributions
-const MAX_NUM_LEVELS_TO_SHOW_BARS = 12
 
 import Distributions: pdf, logpdf, support, mode
 
@@ -35,4 +33,9 @@ export pdf, logpdf, support, mode
 # re-export from ScientificTypesBase:
 export Multiclass, OrderedFactor
 
+# for julia < 1.9
+if !isdefined(Base, :get_extension)
+  include("../ext/UnivariateFiniteDisplayExt.jl")
+end
+
 end

src/methods.jl

@@ -82,23 +82,6 @@ end
 Base.show(io::IO, mime::MIME"text/plain",
           d::UnivariateFinite) = show(io, d)
 
-# in common case of `Real` probabilities we can do a pretty bar plot:
-function Base.show(io::IO, mime::MIME"text/plain",
-                   d::UnivariateFinite{<:Finite{K},V,R,P}) where {K,V,R,P<:Real}
-    show_bars = false
-    if K <= MAX_NUM_LEVELS_TO_SHOW_BARS &&
-        all(>=(0), values(d.prob_given_ref))
-        show_bars = true
-    end
-    show_bars || return show(io, d)
-    s = support(d)
-    x = string.(CategoricalArrays.DataAPI.unwrap.(s))
-    y = pdf.(d, s)
-    S = d.scitype
-    plt = barplot(x, y, title="UnivariateFinite{$S}")
-    show(io, mime, plt)
-end
-
 show_prefix(u::UnivariateFiniteArray{S,V,R,P,1}) where {S,V,R,P} =
     "$(length(u))-element"
 show_prefix(u::UnivariateFiniteArray) = join(size(u),'x')

test/methods.jl

@@ -268,6 +268,20 @@ end
     # @test v ≈ v_close
 end
 
+function displays_okay(v) # `v` a "probability" vector
+    d = UnivariateFinite(v, pool=missing);
+    str = sprint(show, "text/plain", d)
+    contains(str, "UnivariateFinite{Multiclass{$(length(v))}}")
+end
+
+@testset "display" begin
+    @test displays_okay([0.3, 0.7])
+    @test displays_okay([5 + 3im, 4 - 7im])
+    using UnicodePlots
+    @test displays_okay([0.3, 0.7])
+    @test displays_okay([5 + 3im, 4 - 7im])
+end
+
 end # module
 
 true