Make ReactionModelProbPolicy a bit safer

Co-authored-by: Krzysztof Maziarz <krzysztof.maziarz@microsoft.com>
microsoft · Sep 14, 2023 · fbf67b0 · fbf67b0
1 parent df84065
commit fbf67b0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/syntheseus/search/node_evaluation/common.py b/syntheseus/search/node_evaluation/common.py
@@ -50,5 +50,7 @@ def _get_reaction(self, node: Union[MolSetNode, AndNode], graph) -> BackwardReac
             parents = list(graph.predecessors(node))
             assert len(parents) == 1, "Graph must be a tree"
             return graph._graph.edges[parents[0], node]["reaction"]
-        else:
+        elif isinstance(node, AndNode):
             return node.reaction
+        else:
+            raise ValueError(f"ReactionModelProbPolicy does not support nodes of type {type(node)}")