Import error

[mberr]

When trying to run the example from the README, I currently get the following error

Traceback (most recent call last):
  File ".../torch_max_mem/tmp.py", line 2, in <module>
    from torch_max_mem import maximize_memory_utilization
ModuleNotFoundError: No module named 'torch_max_mem'

When I check pip list, the package name appears to be the stylized name

$ pip list | grep max
torch-max-mem     0.0.1.dev0 .../torch_max_mem/src

[cthoyt]

In b3470ff I added a unit test which seems to run alright (though the test currently fails ;) , maybe there's something funny with the way your installation/environment is working

[mberr]

I was in the wrong Python venv 😅