diff --git a/pykat/node_network.py b/pykat/node_network.py index b1bb13a9cef44e2a99c5743d30cfef892f93da42..0db121648421a08f63a326b2f2de0f4b4a990412 100644 --- a/pykat/node_network.py +++ b/pykat/node_network.py @@ -12,7 +12,7 @@ if USE_GUI: import pykat.exceptions as pkex -from pykat.components import Component +from pykat.components import Component, NodeGaussSetter from pykat.detectors import BaseDetector as Detector from pykat.utilities.optics.gaussian_beams import beam_param @@ -65,7 +65,7 @@ class NodeNetwork(object): node_old is the node that is attached to the component. This will accept str - name of a node pykat.node_network.Node - The direct node object - pykat.components.NodeGaussSetter - the node object that is used to set gaussian parameters + NodeGaussSetter - the node object that is used to set gaussian parameters This will call a components __on_node_change callback function to let it know that the nodes connected to it have changed. @@ -77,10 +77,10 @@ class NodeNetwork(object): if isinstance(node_new, str): node_new = self.__kat.nodes[node_new] - if isinstance(node_old, pykat.components.NodeGaussSetter): + if isinstance(node_old, NodeGaussSetter): node_old = node_old.node - if isinstance(node_new, pykat.components.NodeGaussSetter): + if isinstance(node_new, NodeGaussSetter): node_new = node_new.node if not node_new.isDump and node_new.components.count(None) == 0: @@ -204,14 +204,14 @@ class NodeNetwork(object): node_old is the node that is attached to the component. This will accept str - name of a node pykat.node_network.Node - The direct node object - pykat.components.NodeGaussSetter - the node object that is used to set gaussian parameters + NodeGaussSetter - the node object that is used to set gaussian parameters """ if isinstance(node, str): node = self.__kat.nodes[node] - if isinstance(node, pykat.components.NodeGaussSetter): + if isinstance(node, NodeGaussSetter): node = node.node if not isinstance(node, Node): @@ -399,13 +399,13 @@ class NodeNetwork(object): if isinstance(from_node, str): from_node = self.__kat.nodes[from_node] - if isinstance(from_node, pykat.components.NodeGaussSetter): + if isinstance(from_node, NodeGaussSetter): from_node = from_node.node if isinstance(to_node, str): to_node = self.__kat.nodes[to_node] - if isinstance(to_node, pykat.components.NodeGaussSetter): + if isinstance(to_node, NodeGaussSetter): to_node = to_node.node if to_node == from_node: