[BUGFIX] Import missing module

models/unsupervised.py uses `pt` in line 37, but `pt` is undefined in the file.
I wonder why Python doesn't complain about this. Perhaps because unsupervised.py
is never run in isolation and `pt` is otherwise available in the namespace of
the example scripts that use unsupervised.py.
This commit is contained in:
Jensun Ravichandran 2021-06-02 00:31:57 +02:00
parent 91b57b01b1
commit aa42b9e331

View File

@ -3,6 +3,7 @@
import logging import logging
import warnings import warnings
import prototorch as pt
import pytorch_lightning as pl import pytorch_lightning as pl
import torch import torch
import torchmetrics import torchmetrics