def __enter__(self): self._assert_not_used() assert self._step is None, 'This Task already has an execution step.' from caffe2.python import net_builder self._net_builder = net_builder.NetBuilder() self._net_builder.__enter__() return self
def __enter__(self): # temporarily remove from _tasks_to_add to ensure correct order if self.group is not None: self.group._tasks_to_add.remove(self) self._assert_not_used() assert self._step is None, 'This Task already has an execution step.' from caffe2.python import net_builder self._net_builder = net_builder.NetBuilder(_fullname=self.name) self._net_builder.__enter__() return self