Restore Div and MacroElement

branca/element.py

@@ -583,6 +583,9 @@ def render(self, **kwargs) -> str:
             "You cannot render this Element " "if it is not in a Figure."
         )
 
+        for name, element in self._children.items():
+            element.render(**kwargs)
+
         for name, element in self.header._children.items():
             figure.header.add_child(element, name=name)
 
@@ -601,8 +604,6 @@ def render(self, **kwargs) -> str:
         if script is not None:
             figure.script.add_child(Element(script(self, kwargs)), name=self.get_name())
 
-        return figure.render()
-
     def _repr_html_(self, **kwargs) -> str:
         """Displays the Div in a Jupyter notebook."""
         if self._parent is None:
@@ -733,4 +734,5 @@ def render(self, **kwargs) -> str:
         if script is not None:
             figure.script.add_child(Element(script(self, kwargs)), name=self.get_name())
 
-        return figure.render()
+        for name, element in self._children.items():
+            element.render(**kwargs)