TypeError: register_autograd() takes 2 positional arguments but 3 were given

[jorgedanielrodrividal]

Hi! Great tutorial, thank you for sharing it.

I have exactly the same ops.py and setup.py files you shared in this repo but get the following error

TypeError: register_autograd() takes 2 positional arguments but 3 were given

Any ideas on how to debug this?

Thank you!