From 3d291dea4acd0be1bd2f993a3b1cf8a9f971f3a1 Mon Sep 17 00:00:00 2001
From: thomwolf <thomwolf@gmail.com>
Date: Sun, 4 Nov 2018 21:27:19 +0100
Subject: [PATCH] clean up tests

---
 tests/optimization_test.py | 2 +-
 tests/tokenization_test.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/optimization_test.py b/tests/optimization_test.py
index dacec4099b..1675c8b496 100644
--- 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):
 
diff --git a/tests/tokenization_test.py b/tests/tokenization_test.py
index 1cfc520ca1..8d6ede9300 100644
--- 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):