Update src/active_inference_forager/agents/dqn_fep_agent.py

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>

Author: leonvanbokhorst

src/active_inference_forager/agents/dqn_fep_agent.py

@@ -163,7 +163,9 @@ def update_belief(self, observation: np.ndarray) -> None:
 
     def _update_belief_recursive(self, node: BeliefNode, observation: np.ndarray):
         # Ensure observation is a numpy array of floats
-        observation = np.asarray(observation, dtype=np.float64)
+        observation = np.asarray(observation)
+        if observation.dtype != node.mean.dtype:
+            observation = observation.astype(node.mean.dtype)
 
         prediction_error = observation - node.mean
         node.precision += (