Sets the root model on all subplots.

def set_root(self, root):
        """
        Sets the root model on all subplots.
        """
        if root is None:
            return
        for plot in self.traverse(lambda x: x):
            plot._root = root