From aa42b9e331f7ae46c4655a522aa08862b252cfc6 Mon Sep 17 00:00:00 2001 From: Jensun Ravichandran Date: Wed, 2 Jun 2021 00:31:57 +0200 Subject: [PATCH] [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. --- prototorch/models/unsupervised.py | 1 + 1 file changed, 1 insertion(+) diff --git a/prototorch/models/unsupervised.py b/prototorch/models/unsupervised.py index 235a1a9..3fe0123 100644 --- a/prototorch/models/unsupervised.py +++ b/prototorch/models/unsupervised.py @@ -3,6 +3,7 @@ import logging import warnings +import prototorch as pt import pytorch_lightning as pl import torch import torchmetrics