Merge branch 'dev' into main

Project.toml

@@ -4,7 +4,8 @@ authors = [ "Peter Thestrup Waade [email protected]",
             "Anna Hedvig Møller [email protected]",
             "Jacopo Comoglio [email protected]",
             "Christoph Mathys [email protected]"]
-version = "0.5.3"
+version = "0.5.4"
+
 
 [deps]
 ActionModels = "320cf53b-cc3b-4b34-9a10-0ecb113566a3"

src/utils/get_surprise.jl

@@ -61,6 +61,12 @@ Equation:
 """
 function get_surprise(node::ContinuousInputNode)
 
+    #If there was no input
+    if ismissing(node.states.input_value)
+        #Return no surprise
+        return 0
+    end
+
     #Sum the predictions of the vaue parents
     parents_prediction_mean = 0
     for parent in node.edges.observation_parents
@@ -85,6 +91,12 @@ Calculate the surprise of a binary input node on seeing the last input.
 
 function get_surprise(node::BinaryInputNode)
 
+    #If there was no input
+    if ismissing(node.states.input_value)
+        #Return no surprise
+        return 0
+    end
+
     #Sum the predictions of the vaue parents
     parents_prediction_mean = 0
     for parent in node.edges.observation_parents
@@ -137,6 +149,12 @@ Calculate the surprise of a categorical input node on seeing the last input.
 """
 function get_surprise(node::CategoricalInputNode)
 
+    #If there was no input
+    if ismissing(node.states.input_value)
+        #Return no surprise
+        return 0
+    end
+
     #Get value parent
     parent = node.edges.observation_parents[1]