Merge pull request #190 from JuliaDiff/ox/nostatsbase

stop depending on StatsBase
JuliaDiff · Jul 25, 2023 · bdaeb26 · bdaeb26
2 parents 3df2687 + 070d372
commit bdaeb26
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 56 deletions.
diff --git a/Manifest.toml b/Manifest.toml
diff --git a/Project.toml b/Project.toml
@@ -12,7 +12,6 @@ InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
-StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 StructArrays = "09ab397b-f2b6-538f-b94a-2f83cf4a842a"
 
 [compat]
@@ -23,6 +22,5 @@ Cthulhu = "2"
 OffsetArrays = "1"
 PrecompileTools = "1"
 StaticArrays = "1"
-StatsBase = "0.33"
 StructArrays = "0.6"
 julia = "1.10"
diff --git a/src/higher_fwd_rules.jl b/src/higher_fwd_rules.jl
@@ -1,5 +1,4 @@
 using Combinatorics
-using StatsBase
 
 using Base.Iterators
 

diff --git a/src/jet.jl b/src/jet.jl
@@ -239,6 +239,14 @@ expressions for the t′ᵢ that are hopefully easier on the compiler.
     end...)
 end
 
+function countmap(xs)
+    counts = Dict{eltype(xs), Int}()
+    for x in xs
+        counts[x] = get(counts, x, 0) + 1
+    end
+    return counts
+end
+
 @generated function (j::Jet{T, N} where T)(x::TaylorBundle{M}) where {N, M}
     O = min(M,N)
     quote