[DO NOT MERGE] Debugging ALL_MODELS

.github/workflows/CI.yml

@@ -31,29 +31,29 @@ jobs:
           - version: '1'
             os: ubuntu-latest
             num_threads: 2
-          # Minimum supported version
-          - version: 'min'
-            os: ubuntu-latest
-            num_threads: 2
-          # 1.11
-          - version: '1.11'
-            os: ubuntu-latest
-            num_threads: 2
-          # Single-threaded
-          - version: '1'
-            os: ubuntu-latest
-            num_threads: 1
-          # Windows
+          # # Minimum supported version
+          # - version: 'min'
+          #   os: ubuntu-latest
+          #   num_threads: 2
+          # # 1.11
+          # - version: '1.11'
+          #   os: ubuntu-latest
+          #   num_threads: 2
+          # # Single-threaded
+          # - version: '1'
+          #   os: ubuntu-latest
+          #   num_threads: 1
+          # # Windows
           - version: '1'
             os: windows-latest
             num_threads: 2
-          # macOS
-          - version: '1'
-            os: macos-latest
-            num_threads: 2
+          # # macOS
+          # - version: '1'
+          #   os: macos-latest
+          #   num_threads: 2
         test_group:
           - Group1
-          - Group2
+          # - Group2
 
     steps:
       - uses: actions/checkout@v6

docs/src/api.md

@@ -267,6 +267,7 @@ DynamicPPL provides several demo models in the `DynamicPPL.TestUtils` submodule.
 
 ```@docs
 DynamicPPL.TestUtils.DEMO_MODELS
+DynamicPPL.TestUtils.ALL_MODELS
 ```
 
 For every demo model, one can define the true log prior, log likelihood, and log joint probabilities.

src/abstract_varinfo.jl

@@ -838,10 +838,14 @@ function link!!(
     t::StaticTransformation{<:Bijectors.Transform}, vi::AbstractVarInfo, ::Model
 )
     b = inverse(t.bijector)
+    @show b
     x = vi[:]
+    @show x
     y, logjac = with_logabsdet_jacobian(b, x)
+    @show y, logjac
     # Set parameters and add the logjac term.
     vi = unflatten(vi, y)
+    @show vi[:]
     if hasacc(vi, Val(:LogJacobian))
         vi = acclogjac!!(vi, logjac)
     end
@@ -866,7 +870,7 @@ end
 function link(vi::AbstractVarInfo, vns::VarNameTuple, model::Model)
     return link(default_transformation(model, vi), vi, vns, model)
 end
-function link(t::DynamicTransformation, vi::AbstractVarInfo, model::Model)
+function link(t::AbstractTransformation, vi::AbstractVarInfo, model::Model)
     return link!!(t, deepcopy(vi), model)
 end
 
@@ -932,7 +936,7 @@ end
 function invlink(vi::AbstractVarInfo, vns::VarNameTuple, model::Model)
     return invlink(default_transformation(model, vi), vi, vns, model)
 end
-function invlink(t::DynamicTransformation, vi::AbstractVarInfo, model::Model)
+function invlink(t::AbstractTransformation, vi::AbstractVarInfo, model::Model)
     return invlink!!(t, deepcopy(vi), model)
 end
 

src/test_utils/model_interface.jl

@@ -92,7 +92,9 @@ Even though it is recommended to implement this by hand for a particular `Model`
 a default implementation using [`SimpleVarInfo{<:Dict}`](@ref) is provided.
 """
 function varnames(model::Model)
-    return collect(keys(last(DynamicPPL.init!!(model, SimpleVarInfo(Dict())))))
+    result = collect(keys(last(DynamicPPL.init!!(model, SimpleVarInfo(Dict())))))
+    # Concretise the element type.
+    return [x for x in result]
 end
 
 """