also in Multinomial, add CategoricalArrays in tests

JuliaStats · Dec 5, 2020 · 0506b46 · 0506b46
1 parent 726b25c
commit 0506b46
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -20,6 +20,7 @@ julia = "1.3"
 
 [extras]
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"
 
 [targets]
-test = ["Test"]
+test = ["Test","CategoricalArrays"]
diff --git a/src/Multinomial.jl b/src/Multinomial.jl
@@ -123,7 +123,7 @@ end
 
 macro check_and_return_multi()
     esc(quote
-        check_jerr(jerr[1], maxit)
+        check_jerr(jerr[1], maxit,pmax)
         lmu = lmu[1]
         # first lambda is infinity; changed to entry point
         if isempty(lambda) && length(alm) > 2

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,5 +1,5 @@
 using GLMNet, Distributions
-using Random, DataFrames, SparseArrays, Test
+using Random, DataFrames, SparseArrays, Test, CategoricalArrays
 
 X = [74    1  93  93  79  18
      98   36   2  27  65  70