Refactor Invoker/Tracer class to accept Any instead of Tuple[Any] as *inputs - Githubissues

ndif-team / nnsight

The nnsight package enables interpreting and manipulating the internals of deep learned models.

https://nnsight.net/

MIT License

400 stars 37 forks source link

Refactor Invoker/Tracer class to accept Any instead of Tuple[Any] as *inputs #94

Closed koayon closed 7 months ago

koayon commented 7 months ago

Hello! 👋

This tiny PR refactors the Invoker class in order to change the *inputs parameter to accept Any instead of Tuple[Any]. (Discussed with JadenFK in NDIF discord server)

Additionally, it uses conftest to ignore the example files which start with the prefix test.

These files don't seem to be pytest files but were being picked up due to their naming convention. There are other solutions including renaming the files but this approach seemed the least invasive, happy to take this part out of the PR if you have another approach in mind though!

There are no changes to functionality and all the tests are green 👌

Thanks @JadenFiotto-Kaufman!