def __exit__()

in src/optimum/nvidia/utils/patching.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

    def __exit__(self, exc_type, exc_value, traceback):
        with open(self.builder_path, "r") as file:
            lines = file.readlines()

        for i in range(len(lines)):
            if lines[i] == "        optimize(network)\n":
                lines[i - 1] = ""
                lines[i - 2] = ""
                lines[i] = "    optimize(network)\n"
                lines[i + 1] = ""
                lines[i + 2] = ""

        with open(self.builder_path, "w") as f:
            f.writelines(lines)