    def set_type(self, graph_type):
        """Set the graph's type, 'graph' or 'digraph'."""
        self.obj_dict["type"] = graph_type
Set the graph's type, 'graph' or 'digraph'.
python/aitemplate/utils/visualization/pydot.py