I used ACDC on this custom task, does it make sense to merge?

[Pratik-Doshi-99]

I am trying to find a circuit for this custom "iterate" task. To use ACDC, I added a new module: Automatic-Circuit-Discovery/acdc/iterate with a utils.py file. I also configured main.py to call this module and run ACDC on its data.

Task prompt: "def iterate(arr):\n\t" (total 8 tokens)
Corrupted prompt: "def [A]B:\n\t" (total 8 tokens)

I expect "for" (as in for loop) to be the next token. I used ACDC, and it ran successfully.
Target model: GPT2-Neo 125M

Here is the implementation: https://github.com/Pratik-Doshi-99/Automatic-Circuit-Discovery

Does it make sense to merge this to the main repo?