Fixes for imports removed by PR: astropy/astropy#12633

spacetelescope · Jan 3, 2022 · 911b732 · 911b732
1 parent 2640d89
commit 911b732
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/poppy/tests/test_multiprocessing.py b/poppy/tests/test_multiprocessing.py
@@ -10,7 +10,6 @@
 import astropy.io.fits as fits
 import sys
 from distutils.version import LooseVersion
-from astropy.tests.helper import remote_data
 
 try:
     import pytest

diff --git a/poppy/utils.py b/poppy/utils.py
@@ -1340,7 +1340,7 @@ def __init__(self, func=None, **kwargs):
         self.decorator_kwargs = kwargs
 
     def __call__(self, wrapped_function):
-        from astropy.utils.decorators import wraps
+        from functools import wraps
         from astropy.units import UnitsError, add_enabled_equivalencies, Quantity
         import inspect