fix: cpd floating point comparison issue (#172)

* fix: cpd floating point comparison issue * fix: imports ordering
mckinsey · Sep 13, 2022 · c42d0e2 · c42d0e2
1 parent 4262609
commit c42d0e2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/causalnex/inference/inference.py b/causalnex/inference/inference.py
@@ -32,6 +32,7 @@
 """
 import copy
 import inspect
+import math
 import re
 import types
 from typing import Any, Callable, Dict, Hashable, List, Optional, Tuple, Union
@@ -205,7 +206,7 @@ def _do(self, observation: str, state: Dict[Hashable, float]):
         Raises:
             ValueError: if states do not match original states of the node, or probabilities do not sum to 1.
         """
-        if sum(state.values()) != 1.0:
+        if not math.isclose(sum(state.values()), 1.0):
             raise ValueError("The cpd for the provided observation must sum to 1")
 
         if max(state.values()) > 1.0 or min(state.values()) < 0: