From 0be1d7621505c5cbcab5c77764586e0cc52ba512 Mon Sep 17 00:00:00 2001
From: Javier Sanchez <jsanchez@Javiers-MacBook-Air.local>
Date: Tue, 20 Feb 2024 16:14:14 -0500
Subject: [PATCH] flake8 fixes

---
 augur/generate.py | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/augur/generate.py b/augur/generate.py
index b3c2ec6..cd95c4c 100644
--- a/augur/generate.py
+++ b/augur/generate.py
@@ -209,7 +209,7 @@ def generate_sacc_and_stats(config):
             sources[sacc_tracer].bias = bias
             sys_params[f'{sacc_tracer}_bias'] = bias
             sys_params[f'{sacc_tracer}_delta_z'] = delta_z[i]
-    
+
     # Read data vector combinations
     if 'statistics' not in config.keys():
         raise ValueError('statistics key is required in config file')
@@ -236,11 +236,11 @@ def generate_sacc_and_stats(config):
             wgt = np.zeros((len(ells_aux), len(ells_here)))
             for i in range(len(ells_here)):
                 in_win = (ells_aux > ell_edges[i]) & (ells_aux < ell_edges[i+1])
-                wgt[in_win, i] = 1.0 
+                wgt[in_win, i] = 1.0
             win = sacc.BandpowerWindow(ells_aux, wgt)
             S.add_ell_cl(key, tr1, tr2,
                          ells_here, np.zeros(len(ells_here)), window=win)
-            
+
             # Now create TwoPoint objects for firecrown
             _aux_stat = TwoPoint(source0=sources[tr1], source1=sources[tr2],
                                  sacc_data_type=key)
@@ -265,20 +265,19 @@ def generate(config, return_all_outputs=False, write_sacc=True):
         likelihood object.
     write_sacc : bool
         If `True` it writes a sacc file with fiducial data vector.
-    
 
     Returns:
     -------
     lk : firecrown.likelihood.ConstGaussian
         Likelihood object, only returned if `return_all_outputs` is True.
     S : sacc.Sacc
-        Sacc object with fake data vector and covariance. It is only returned if `return_all_outputs` 
-        is True.
+        Sacc object with fake data vector and covariance. It is only returned if
+        `return_all_outputs` is True.
     tools : firecrown.modeling.ModelingTools
         Modeling tools, only returned if `return_all_outputs` is True.
     sys_params : dict
-        Dictionary containing the modeling systematic parameters. It is only returned if `return_all_outputs`
-        is True.
+        Dictionary containing the modeling systematic parameters. It is only returned if
+        `return_all_outputs` is True.
 
     """