clean up tests

2018-11-04 21:27:19 +01:00
parent 87da161c2a
commit 3d291dea4a
2 changed files with 2 additions and 2 deletions
--- a/tests/optimization_test.py
+++ b/tests/optimization_test.py
@@ -20,7 +20,7 @@ import unittest

 import torch

-import optimization as optimization
+import optimization

 class OptimizationTest(unittest.TestCase):

--- a/tests/tokenization_test.py
+++ b/tests/tokenization_test.py
@@ -19,7 +19,7 @@ from __future__ import print_function
 import os
 import unittest

-import tokenization as tokenization
+import tokenization


 class TokenizationTest(unittest.TestCase):