ruff fixes

RadioAstronomySoftwareGroup · Aug 22, 2024 · bb22559 · bb22559
1 parent d2c49c7
commit bb22559
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 13 deletions.
diff --git a/src/pyuvdata/uvbeam/__init__.py b/src/pyuvdata/uvbeam/__init__.py
@@ -2,6 +2,7 @@
 # Licensed under the 2-clause BSD License
 
 """Init file for UVBeam."""
+
 from .analytic_beam import AiryBeam, GaussianBeam, ShortDipoleBeam, UniformBeam  # noqa
 from .beam_interface import BeamInterface  # noqa
 from .uvbeam import *  # noqa
diff --git a/src/pyuvdata/uvbeam/analytic_beam.py b/src/pyuvdata/uvbeam/analytic_beam.py
@@ -1,8 +1,8 @@
-# -*- mode: python; coding: utf-8 -*-
 # Copyright (c) 2022 Radio Astronomy Software Group
 # Licensed under the 2-clause BSD License
 
 """Analytic beam class definitions."""
+
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
@@ -113,8 +113,7 @@ def __iter__(self) -> str:
             for a in dir(self)
             if not a.startswith("_") and not callable(getattr(self, a))
         ]
-        for a in attribute_list:
-            yield a
+        yield from attribute_list
 
     def __eq__(self, other: Any, silent: bool = False) -> bool:
         """Equality method."""
@@ -834,7 +833,7 @@ def _power_eval(
 
         if self.Npols > self.Nfeeds:
             # cross pols are included
-            data_array[0, 2] = -np.sin(za_fgrid) ** 2 * np.sin(2.0 * az_fgrid) / 2.0
+            data_array[0, 2] = -(np.sin(za_fgrid) ** 2) * np.sin(2.0 * az_fgrid) / 2.0
             data_array[0, 3] = data_array[0, 2]
 
         return data_array

diff --git a/src/pyuvdata/uvbeam/beam_interface.py b/src/pyuvdata/uvbeam/beam_interface.py
@@ -1,8 +1,8 @@
-# -*- mode: python; coding: utf-8 -*-
 # Copyright (c) 2022 Radio Astronomy Software Group
 # Licensed under the 2-clause BSD License
 
 """Definition for BeamInterface object."""
+
 from __future__ import annotations
 
 import copy

diff --git a/tests/uvbeam/test_analytic_beam.py b/tests/uvbeam/test_analytic_beam.py
@@ -1,4 +1,3 @@
-# -*- mode: python; coding: utf-8 -*-
 # Copyright (c) 2024 Radio Astronomy Software Group
 # Licensed under the 2-clause BSD License
 
@@ -202,8 +201,8 @@ def test_short_dipole_beam(az_za_deg_grid):
 
     expected_data[0, 0] = 1 - np.sin(za_vals) ** 2 * np.cos(az_vals) ** 2
     expected_data[0, 1] = 1 - np.sin(za_vals) ** 2 * np.sin(az_vals) ** 2
-    expected_data[0, 2] = -np.sin(za_vals) ** 2 * np.sin(2.0 * az_vals) / 2.0
-    expected_data[0, 3] = -np.sin(za_vals) ** 2 * np.sin(2.0 * az_vals) / 2.0
+    expected_data[0, 2] = -(np.sin(za_vals) ** 2) * np.sin(2.0 * az_vals) / 2.0
+    expected_data[0, 3] = -(np.sin(za_vals) ** 2) * np.sin(2.0 * az_vals) / 2.0
 
     np.testing.assert_allclose(power_vals, expected_data)
 

diff --git a/tests/uvbeam/test_beam_interface.py b/tests/uvbeam/test_beam_interface.py
@@ -1,4 +1,3 @@
-# -*- mode: python; coding: utf-8 -*-
 # Copyright (c) 2024 Radio Astronomy Software Group
 # Licensed under the 2-clause BSD License
 import copy
@@ -93,10 +92,7 @@ def test_beam_interface(
     nfreqs = 20
     freq_array = np.linspace(100e6, 150e6, nfreqs)
 
-    if "include_cross_pols" in kwargs.keys():
-        include_cross_pols = kwargs["include_cross_pols"]
-    else:
-        include_cross_pols = True
+    include_cross_pols = kwargs.get("include_cross_pols", True)
 
     analytic = beam_obj(**kwargs)