Skip to content

Commit a325a3b

Browse files
committed
Fix import errors in unitests.
1 parent dd41a12 commit a325a3b

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

deep_speech_2/models/tests/test_decoders.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from __future__ import print_function
55

66
import unittest
7-
from decoder import *
7+
from models import decoder
88

99

1010
class TestDecoders(unittest.TestCase):
@@ -53,11 +53,13 @@ def setUp(self):
5353
self.beam_search_result = ['acdc', "b'a"]
5454

5555
def test_greedy_decoder_1(self):
56-
bst_result = ctc_greedy_decoder(self.probs_seq1, self.vocab_list)
56+
bst_result = decoder.ctc_greedy_decoder(self.probs_seq1,
57+
self.vocab_list)
5758
self.assertEqual(bst_result, self.greedy_result[0])
5859

5960
def test_greedy_decoder_2(self):
60-
bst_result = ctc_greedy_decoder(self.probs_seq2, self.vocab_list)
61+
bst_result = decoder.ctc_greedy_decoder(self.probs_seq2,
62+
self.vocab_list)
6163
self.assertEqual(bst_result, self.greedy_result[1])
6264

6365
def test_beam_search_decoder_1(self):

deep_speech_2/utils/tests/test_error_rate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from __future__ import print_function
66

77
import unittest
8-
import error_rate
8+
from utils import error_rate
99

1010

1111
class TestParse(unittest.TestCase):

0 commit comments

Comments
 (0)