[FIX][CLEANUP] Module import, remove version testing in code

[evelynmitchell]

Tests were failing with a bad import:

from torch import Module

in freeze layers.

There were also several instances of checking for and handling old version of pytorch, which I removed now that dependencies are handled by poetry.

[evelynmitchell]

Exposed seek_all_images to fix erroring tests.