Fix isnan for NamedTuple distributions

Author: penelopeysm

HISTORY.md

@@ -1,5 +1,9 @@
 # DynamicPPL Changelog
 
+## 0.35.9
+
+Fixed the `isnan` check introduced in 0.35.7 for distributions which returned NamedTuple.
+
 ## 0.35.8
 
 Added the `DynamicPPL.TestUtils.AD.run_ad` function to test the correctness and/or benchmark the performance of an automatic differentiation backend on DynamicPPL models.

Project.toml

@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.35.8"
+version = "0.35.9"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

src/debug_utils.jl

@@ -242,6 +242,10 @@ function _has_missings(x::AbstractArray)
     return false
 end
 
+_has_nans(x::NamedTuple) = any(_has_nans, x)
+_has_nans(x::AbstractArray) = any(_has_nans, x)
+_has_nans(x) = isnan(x)
+
 # assume
 function record_pre_tilde_assume!(context::DebugContext, vn, dist, varinfo)
     record_varname!(context, vn, dist)
@@ -291,7 +295,7 @@ function record_pre_tilde_observe!(context::DebugContext, left, dist, varinfo)
         )
     end
     # Check for NaN's as well
-    if any(isnan, left)
+    if _has_nans(left)
         error(
             "Encountered a NaN value on the left-hand side of an" *
             " observe statement; this may indicate that your data" *