Index: leahi_dialin/td/treatment_delivery.py =================================================================== diff -u -rd3f64754a55d930463ad518acc34931e9a3867f1 -r8c823851a48f72a6ac2a26cc691063d214288231 --- leahi_dialin/td/treatment_delivery.py (.../treatment_delivery.py) (revision d3f64754a55d930463ad518acc34931e9a3867f1) +++ leahi_dialin/td/treatment_delivery.py (.../treatment_delivery.py) (revision 8c823851a48f72a6ac2a26cc691063d214288231) @@ -22,6 +22,7 @@ from .modules.bubble_detector import TDBubbleDectector from .modules.buttons import TDButtons from .modules.constants import NO_RESET, RESET +from .modules.ejector import TDEjector from .modules.pressure_sensors import TDPressureSensors from .modules.switches import TDSwitches from .modules.treatment import TDTreatment @@ -122,6 +123,7 @@ self.blood_flow = TDBloodFlow(self.can_interface, self.logger) self.bubbles = TDBubbleDectector(self.can_interface, self.logger) self.buttons = TDButtons(self.can_interface, self.logger) + self.ejector = TDEjector(self.can_interface, self.logger) self.pressure_sensors = TDPressureSensors(self.can_interface, self.logger) self.switches = TDSwitches(self.can_interface, self.logger) self.treatment = TDTreatment(self.can_interface, self.logger)