Check the input length in logdensity

src/logdensityfunction.jl

@@ -529,6 +529,13 @@ end
 @inline function LogDensityProblems.logdensity(
     ldf::LogDensityFunction, params::AbstractVector{<:Real}
 )
+    if length(params) != ldf._dim
+        throw(
+            ArgumentError(
+                "The length of the input vector is $(length(params)), but the LogDensityFunction expects a vector of length $(ldf._dim) based on the ranges that were extracted when the LogDensityFunction was constructed.",
+            ),
+        )
+    end
     return logdensity_internal(
         params,
         ldf.model,

test/logdensityfunction.jl

@@ -180,6 +180,18 @@ end
         @test LogDensityProblems.dimension(ldf) == 4
     end
 
+    @testset "logdensity checks the input length" begin
+        @model function m3()
+            x ~ Normal()
+            y ~ Normal()
+            return nothing
+        end
+        ldf = DynamicPPL.LogDensityFunction(m3())
+        @test LogDensityProblems.logdensity(ldf, zeros(2)) isa Real
+        @test_throws ArgumentError LogDensityProblems.logdensity(ldf, zeros(1))
+        @test_throws ArgumentError LogDensityProblems.logdensity(ldf, zeros(3))
+    end
+
     @testset "capabilities" begin
         @model f() = x ~ Normal()
         model = f()