made Ruff happy again

punch-mission · Nov 12, 2024 · 6f3fc65 · 6f3fc65
1 parent c5dda5c
commit 6f3fc65
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/simpunch/level2.py b/simpunch/level2.py
@@ -140,7 +140,7 @@ def generate_starfield(wcs: WCS,
 
 def generate_dummy_polarization(map_scale: float = 0.225,
                                 pol_factor: float = 0.5) -> NDCube:
-    """Create a synthetic polarization map"""
+    """Create a synthetic polarization map."""
     shape = [int(floor(180 / map_scale)), int(floor(360 / map_scale))]
     xcoord = np.linspace(-pol_factor, pol_factor, shape[1])
     ycoord = np.linspace(-pol_factor, pol_factor, shape[0])

diff --git a/simpunch/tests/test_starfield.py b/simpunch/tests/test_starfield.py
@@ -34,12 +34,11 @@ def _sample_ndcube(shape: tuple, code:str = "PM1", level:str = "0") -> NDCube:
 @pytest.fixture
 def sample_ndcollection() -> NDCollection:
     input_data = sample_ndcube((2048, 2048))
-    sample_collection = NDCollection(
+    return NDCollection(
             [("-60.0 deg", input_data),
              ("0.0 deg", input_data),
              ("60.0 deg", input_data)],
             aligned_axes="all")
-    return sample_collection
 
 def test_starfield(sample_ndcube: NDCube) -> None:
     """Test starfield generation."""