diff options
Diffstat (limited to 'beliefs/types/BernoulliOrNode.py')
-rw-r--r-- | beliefs/types/BernoulliOrNode.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/beliefs/types/BernoulliOrNode.py b/beliefs/types/BernoulliOrNode.py index 27da85a..ce497b9 100644 --- a/beliefs/types/BernoulliOrNode.py +++ b/beliefs/types/BernoulliOrNode.py @@ -6,7 +6,7 @@ from beliefs.types.Node import ( MessageType, InvalidLambdaMsgToParent ) -from beliefs.factors.BernoulliOrFactor import BernoulliOrFactor +from beliefs.factors.BernoulliOrCPD import BernoulliOrCPD class BernoulliOrNode(Node): @@ -18,7 +18,7 @@ class BernoulliOrNode(Node): children=children, parents=parents, cardinality=2, - cpd=BernoulliOrFactor(label_id, parents)) + cpd=BernoulliOrCPD(label_id, parents)) def compute_pi_agg(self): if not self.parents: |