Fix failing example

2021-05-21 18:54:47 +02:00
parent 2aa631f4e6
commit e7e6bf9173
1 changed files with 1 additions and 2 deletions
--- a/examples/glvq_spiral.py
+++ b/examples/glvq_spiral.py
@@ -4,8 +4,7 @@ import argparse

 import prototorch as pt
 import pytorch_lightning as pl
-
-#import torch
+import torch

 if __name__ == "__main__":
    # Command-line arguments