Skip to content

GitLab

Unverified Commit 963224f5 authored by Vincent QB's avatar Vincent QB Committed by GitHub
Browse files

remove * in import of models (#932) 

* remove * in import of models.

* only importing WaveRNN in tests.
parent 1df9e201
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 10 deletions
test/torchaudio_unittest/models_test.py
View file @ 963224f5
......@@ -2,15 +2,9 @@ import itertools
from collections import namedtuple
import torch
from torchaudio.models import (
Wav2Letter,
MelResNet,
UpsampleNetwork,
WaveRNN,
ConvTasNet,
)
from parameterized import parameterized
from torchaudio.models import ConvTasNet, Wav2Letter, WaveRNN
from torchaudio.models.wavernn import MelResNet, UpsampleNetwork
from torchaudio_unittest import common_utils
......
torchaudio/models/__init__.py
View file @ 963224f5
from .wav2letter import *
from .wavernn import *
from .wav2letter import Wav2Letter
from .wavernn import WaveRNN
from .conv_tasnet import ConvTasNet
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment