contextmanager didn't move, IDE was confused

django-tastypie · Dec 11, 2019 · 733ecf5 · 733ecf5
1 parent ecfcba5
commit 733ecf5
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/testcases.py b/tests/testcases.py
@@ -1,15 +1,12 @@
 import json
+
+from contextlib import contextmanager
+
 from django.test.testcases import TestCase
 from django.test.utils import CaptureQueriesContext
 from django.db import connections
 
 
-try:
-    from contextlib import contextmanager
-except ImportError:
-    from decorator import contextmanager
-
-
 class TastyPieTestCase(TestCase):
 
     @contextmanager