Merge pull request #160 from ilabcode/dev

Versio 0.5.4

Author: PTWaade

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

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