Merge pull request #160 from ilabcode/dev

Versio 0.5.4
ComputationalPsychiatry · Aug 9, 2024 · 1e1318a · 1e1318a · PTWaade · Aug 9, 2024
2 parents 5e2da94 + b1a24b0
commit 1e1318a
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 2 deletions.
diff --git a/src/ActionModels_variations/utils/reset.jl b/src/ActionModels_variations/utils/reset.jl
@@ -79,8 +79,8 @@ function reset_state!(node::CategoricalStateNode)
 
     node.states.posterior .= missing
     node.states.value_prediction_error .= missing
-    node.states.prediction .= missing
-    node.states.parent_predictions .= missing
+    node.states.prediction .= 1/length(node.states.prediction)
+    node.states.parent_predictions .= 1/length(node.states.parent_predictions)
 
     return nothing
 end

diff --git a/test/testsuite/test_get_surprise.jl b/test/testsuite/test_get_surprise.jl
@@ -0,0 +1,28 @@
+using Test
+using HierarchicalGaussianFiltering
+
+
+@testset "Test get_surprise" begin
+
+    @testset "Test for state-transition HGF" begin
+
+        #Set up test inputs
+        test_inputs = [
+            missing missing 2 missing
+            missing 1 missing missing
+            missing missing missing 3
+            missing missing missing missing
+            3 missing missing missing
+        ]
+
+        #Initialize HGF
+        HGF_test = premade_hgf("categorical_state_transitions", verbose = false)
+
+        #Give inputs
+        give_inputs!(HGF_test, test_inputs)
+
+        #Get surprise
+        @test get_surprise(HGF_test) isa Real
+
+    end
+end