feat: complete q7

2024-05-07 21:30:27 +08:00 · 2024-05-07 21:30:27 +08:00 · 3d47ae1a15
commit 3d47ae1a15
parent cd9629462e
1 changed files with 7 additions and 1 deletions
--- a/inference.py
+++ b/inference.py
@ -606,7 +606,13 @@ class ExactInference(InferenceModule):
        current position is known.
        """
        "*** YOUR CODE HERE ***"
-        raiseNotDefined()
+        new_beliefs = DiscreteDistribution()
+        for old_position in self.allPositions:
+            new_positions = self.getPositionDistribution(gameState, old_position)
+            for new_position, prob in new_positions.items():
+                new_beliefs[new_position] += self.beliefs[old_position] * prob
+        self.beliefs = new_beliefs
+
        "*** END YOUR CODE HERE ***"

    def getBeliefDistribution(self):