pipelines.py 文件源码

def plot(self, file_name=None):
        """
        Plot the DAG of the pipeline
        """
        import matplotlib.pyplot as plt
        nx.draw_networkx(self.dag)
        if not file_name:
            file_name = os.path.join(self.output_dir, self.name + '.png')
        plt.savefig(file_name)