fixing linting issues

spacetelescope · Aug 17, 2021 · d30de97 · d30de97
1 parent d9a7d79
commit d30de97
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/jdaviz/core/data_formats.py b/jdaviz/core/data_formats.py
@@ -4,7 +4,7 @@
 
 import astropy.io
 from specutils.io.registers import identify_spectrum_format
-from specutils import Spectrum1D, SpectrumList
+from specutils import SpectrumList
 
 from jdaviz.core.config import list_configurations
 

diff --git a/jdaviz/tests/test_data_formats.py b/jdaviz/tests/test_data_formats.py
@@ -41,10 +41,10 @@ def create_jwst(n_dim=None, multi=None, ext=None):
         exts.append(ex1d)
         # create two more extensions
         if multi:
-            for i in range(1,3):
+            for i in range(1, 3):
                 ex1d = _create_bintable(ext, ver=i)
                 exts.append(ex1d)
-    shape = (1,) if n_dim == 1 else (1, 2,) if n_dim == 2 else (1, 2, 3) if n_dim == 3 else (1,)
+    shape = (1,) if n_dim == 1 else (1, 2,) if n_dim == 2 else (1, 2, 3) if n_dim == 3 else (1, )
     sci.data = np.empty(shape)
     if n_dim > 1:
         exts.append(sci)