[QA] Add more pre commit checks

This commit is contained in:
Alexander Engelsberger
2021-06-16 13:46:09 +02:00
committed by Alexander Engelsberger
parent d0ae94f2af
commit 11cfa79746
17 changed files with 66 additions and 38 deletions

View File

@@ -1,8 +1,9 @@
"""ProtoTorch components test suite."""
import prototorch as pt
import torch
import prototorch as pt
def test_labcomps_zeros_init():
protos = torch.zeros(3, 2)

View File

@@ -4,6 +4,7 @@ import unittest
import numpy as np
import torch
from prototorch.functions import (activations, competitions, distances,
initializers, losses, pooling)