[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:
parent
91b57b01b1
commit
aa42b9e331
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user