def dump()

in log_analyzer/log.py [0:0]

• 18 lines of code
• 6 McCabe index (conditional complexity)

    def dump(self, filename):
        default_comm_folder_path = "results/comm_logs/"
        if not os.path.exists(default_comm_folder_path):
            os.makedirs(default_comm_folder_path, exist_ok=True)
        if "." in filename:
            filename = filename.split(".")[0]
        filename = os.path.join("results/comm_logs/", filename)
        csv_filename = filename + "_log.csv"
        with open(csv_filename, "w") as f:
            f.write(self.comm_logs[0].csv_header() + "\n")
            for log_item in self.comm_logs:
                log_item_write = copy.deepcopy(log_item)
                if(log_item_write.comm_type == CommType.computation):
                    msg_size_str = "("+' '.join(str(shape).replace(',', '') for shape in log_item_write.msg_size)+")"
                    log_item_write.msg_size = msg_size_str
                f.write(log_item_write.view_as_csv_line() + "\n")
                del log_item_write
        return csv_filename