szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.18k stars 279 forks source link

python 3.12 compatibility #85

Closed leo-ware closed 5 days ago

leo-ware commented 6 months ago

added a github action to automatically run the unit tests

leo-ware commented 6 months ago

What I did:

I added a dependency on looseversion which is a drop in replacement library for the distutils LooseVersion class, which was causing the issue. The only line of code that used it was just there to check the pytorch version and issue a warning, so it might have been possible to remove it. This seemed like the better route though.