MAINT: Fix linting

Author: derb12

pybaselines/two_d/_algorithm_setup.py

@@ -410,19 +410,12 @@ def _setup_whittaker(self, y, lam=1, diff_order=2, weights=None, copy_weights=Fa
         if self._sort_order is not None and weights is not None:
             weight_array = weight_array[self._sort_order]
 
-        if False:#(
-            #TODO can probably keep the basis for reuse if using SVD, like _setup_spline does, and
-            # retain the unmodified penalties for the rows and columns if possible to skip that
-            # calculation as well
-
-            #self.whittaker_system is not None
-            #and self.whittaker_system.same_basis(diff_order, num_eigens)
-        #):
-            self.whittaker_system.update_penalty(lam)
-        else:
-            whittaker_system = WhittakerSystem2D(
-                self._shape, lam, diff_order, num_eigens
-            )
+        # TODO can probably keep the basis for reuse if using SVD, like _setup_spline does, and
+        # retain the unmodified penalties for the rows and columns if possible to skip that
+        # calculation as well
+        whittaker_system = WhittakerSystem2D(
+            self._shape, lam, diff_order, num_eigens
+        )
         if not whittaker_system._using_svd:
             y = y.ravel()
             weight_array = weight_array.ravel()
@@ -607,7 +600,7 @@ def _setup_spline(self, y, weights=None, spline_degree=3, num_knots=10,
         ):
             self._spline_basis = SplineBasis2D(self.x, self.z, num_knots, spline_degree)
 
-        #TODO should probably also retain the unmodified penalties for the rows and
+        # TODO should probably also retain the unmodified penalties for the rows and
         # columns if possible to skip that calculation as well
         pspline = PSpline2D(self._spline_basis, lam, diff_order)
 

tests/test_algorithm_setup.py

@@ -165,7 +165,7 @@ def test_setup_whittaker_reset_penalized_system(small_data, algorithm, diff_orde
     weights = np.linspace(0.01, 1, small_data.shape[0])
     whittaker_system.add_diagonal(weights)
 
-    #TODO update these tests once the original diagonals are tracked by _Algorithm again
+    # TODO update these tests once the original diagonals are tracked by _Algorithm again
     # now reset with a new lam value
     _, _, whittaker_system_2 = algorithm._setup_whittaker(
         small_data, lam_2, diff_order, allow_lower=allow_lower, reverse_diags=reverse_diags