[QA] Add more pre commit checks
This commit is contained in:
committed by
Alexander Engelsberger
parent
d0ae94f2af
commit
11cfa79746
@@ -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)
|
||||
|
@@ -4,6 +4,7 @@ import unittest
|
||||
|
||||
import numpy as np
|
||||
import torch
|
||||
|
||||
from prototorch.functions import (activations, competitions, distances,
|
||||
initializers, losses, pooling)
|
||||
|
||||
|
Reference in New Issue
Block a user