linting

AllenNeuralDynamics · Sep 5, 2024 · 64036b8 · 64036b8
1 parent b06c0a7
commit 64036b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_aind_dynamic_foraging_data_utils.py b/tests/test_aind_dynamic_foraging_data_utils.py
@@ -169,7 +169,7 @@ def test_event_triggered_response_censor(self):
         assert np.isclose(etr_censored.query("time < -1")["y"].mean(), 10, rtol=0.01)
 
         # Test with arbitrary list of censor times
-        censor_times = [2,2.5,3,3.5,4,4.5,5,5.5,6,6.5,7,7.5,8]
+        censor_times = [2, 2.5, 3, 3.5, 4, 4.5, 5, 5.5, 6, 6.5, 7, 7.5, 8]
         etr_censored = alignment.event_triggered_response(
             data=df,
             t="time",
@@ -179,7 +179,7 @@ def test_event_triggered_response_censor(self):
             t_after=2,
             output_sampling_rate=100,
             censor=True,
-            censor_times=censor_times
+            censor_times=censor_times,
         )
 
         # assert properties of etr