def testProofStateEmbedding(self, predictor_str): predictor = self._get_predictor(predictor_str) goal = 'a f bool v A y' state = predictions.ProofState(goal=goal) goal_emb = predictor.goal_embedding(goal) expected_state_emb = predictions.EmbProofState(goal_emb=goal_emb) actual_state_emb = predictor.proof_state_embedding(state) self.assertAllClose(expected_state_emb.goal_emb, actual_state_emb.goal_emb) self.assertAllEqual(expected_state_emb[1:], actual_state_emb[1:])
def proof_state_embedding(
self, state: predictions.ProofState) -> predictions.EmbProofState:
return predictions.EmbProofState(goal_emb=self.goal_embedding(state.goal))
def testProofStateEncoding(self, predictor_str):
predictor = self._get_predictor(predictor_str)
goal_emb = self.embeddings[0]
state_emb = predictions.EmbProofState(goal_emb=goal_emb)
actual_state_enc = predictor.proof_state_encoding(state_emb)
self.assertAllEqual(goal_emb, actual_state_enc)
def proof_state_embedding(self, state: predictions.ProofState):
return predictions.EmbProofState(*[[x.__hash__(), 2] for x in state])
def _proof_state_embedding(predictor, state):
return predictions.EmbProofState(*[[x.__hash__(), 2] for x in state])
