style

SABS-R3-Epidemiology · Mar 4, 2024 · fd99ea3 · fd99ea3
1 parent 5d3a5f9
commit fd99ea3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/epios/post_process_update.py b/epios/post_process_update.py
@@ -659,7 +659,7 @@ def _wrapper_Age_Base(self, sampling_method, sample_size, time_sample,
                 infected_rate = list(ReScaler(false_positive=false_positive,
                                               false_negative=false_negative,
                                               smoothing=smoothing)(observ))
-                
+
             elif sample_strategy == 'Random':  # Change people sampled at each sample time point
 
                 people = []

diff --git a/epios/sampling_maker.py b/epios/sampling_maker.py
@@ -104,7 +104,7 @@ def count_negative(x):
                     return array(obs).sum() * len(x.index) / len(self.data.columns)
                 except ZeroDivisionError:
                     return nan
-        
+
         if keep_track:
             STATUSES = self.data.loc[sampling_times, people]
             result = STATUSES.apply(lambda x: x.apply(self._testresult))

diff --git a/epios/tests/test_post_process_update.py b/epios/tests/test_post_process_update.py
@@ -230,6 +230,7 @@ def test_sampled_non_responder(self):
             np.testing.assert_equal(res[0], [self.range6, [np.nan, 1.0, 1.0, 1.0, np.nan, 1.0]])
         except AssertionError:
             self.fail(str(res[0]) + ' is not ' + str([self.range6, [np.nan, 1.0, 1.0, 1.0, np.nan, 1.0]]))
+
     def test_compare(self):
         _, diff = self.processor.predict.AgeRegion(6, self.range6, comparison=True,
                                                    data_store_path=self.path, gen_plot=True,