From 77699ce7f90dc17a9904dd3c6955e528f4a2100d Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 7 Nov 2022 11:15:30 +0530 Subject: [PATCH] fix tests --- tests/models/complexnn_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/complexnn_test.py b/tests/models/complexnn_test.py index 8c18ed5..524a6cf 100644 --- a/tests/models/complexnn_test.py +++ b/tests/models/complexnn_test.py @@ -1,8 +1,8 @@ import torch from enhancer.models.complexnn.conv import ComplexConv2d, ComplexConvTranspose2d -from enhancer.models.complexnn.norm import ComplexBatchNorm2D from enhancer.models.complexnn.rnn import ComplexLSTM +from enhancer.models.complexnn.utils import ComplexBatchNorm2D def test_complexconv2d(): @@ -12,7 +12,7 @@ def test_complexconv2d(): ) with torch.no_grad(): out = conv(sample_input) - assert out.shape == torch.Size([1, 32, 128, 14]) + assert out.shape == torch.Size([1, 32, 128, 13]) def test_complexconvtranspose2d():