diff --git a/pybn/network.py b/pybn/network.py index e1318a6..1ac903d 100755 --- a/pybn/network.py +++ b/pybn/network.py @@ -466,7 +466,7 @@ def getSize(self): def printNode(self): # print node commentNode = '\t\t\n' - initNode = '\t\t\n' + initNode = '\t\t\n' commentOutcomes = '\t\t\t\n' initOutcomes = '' for outcome in self.outcomes: @@ -477,7 +477,7 @@ def printNode(self): commentArc = '\t\t\t\n' initArc = '\t\t\t' for connection in self.arcConnection: - initArc += connection[0]+' ' + initArc += connection[1]+' ' endArc = '\n' else: commentArc = '' @@ -495,7 +495,7 @@ def printNode(self): return print_node def printExtension(self): - initExtensions = '\t\t\t\n' + initExtensions = '\t\t\t\n' initName = '\t\t\t\t'+self.caption+'\n' initIcolor = '\t\t\t\t\n' initOcolor = '\t\t\t\t\n' @@ -508,7 +508,7 @@ def printExtension(self): return initExtensions+initName+initIcolor+initOcolor+initFont+initPos+initBar+endExtensions def printProbabilities(self): - commentProbabilities = '// setting probabilities for "'+self.name+'"\ntheProbs.Flush();\n' + commentProbabilities = '// setting probabilities for "'+self.nodeId+'"\ntheProbs.Flush();\n' initProbabilities = 'theProbs.SetSize('+str(len(self.probability))+');\n' for i,probability in enumerate(self.probability): initProbabilities += 'theProbs['+str(i)+'] = '+str(probability)+';\n' @@ -622,7 +622,7 @@ class Arc(object): def __init__(self, from_node, to_node): self.from_node = from_node self.to_node = to_node - self.to_node.addArcConnection(self.from_node.getName(),self.from_node.getIdNum(),self.from_node.getSize()) + self.to_node.addArcConnection(self.from_node.getName(),self.from_node.getNodeId(),self.from_node.getSize()) def __repr__(self): return self.name