Index: dialin/hd/hemodialysis_device.py =================================================================== diff -u -r0cab9b7a747d7e2dfc36698ea2481b62971ec786 -rc83af5f3c56db1049d5fde362848a0200279d585 --- dialin/hd/hemodialysis_device.py (.../hemodialysis_device.py) (revision 0cab9b7a747d7e2dfc36698ea2481b62971ec786) +++ dialin/hd/hemodialysis_device.py (.../hemodialysis_device.py) (revision c83af5f3c56db1049d5fde362848a0200279d585) @@ -27,6 +27,7 @@ from .treatment import HDTreatment from .valves import HDValves from .pressure_occlusion import HDPressureOcclusion +from .pretreatment import HDPreTreatment from ..protocols.CAN import (DenaliMessage, DenaliCanMessenger, DenaliChannels) @@ -117,6 +118,7 @@ self.dialysate_inlet_flow = HDDialysateInletFlow(self.can_interface, self.logger) self.dialysate_outlet_flow = HDDialysateOutletFlow(self.can_interface, self.logger) self.treatment = HDTreatment(self.can_interface, self.logger) + self.pretreatment = HDPreTreatment(self.can_interface, self.logger) self.pressure_occlusion = HDPressureOcclusion(self.can_interface, self.logger) self.valves = HDValves(self.can_interface, self.logger)