make yapf

km3io/offline.py

@@ -45,6 +45,7 @@ def fitinf(fitparam, tracks):
         params = fit[count_nested(fit, axis=1) > index]
         return params[:, index]
 
+
 def fitparams():
     """name of the fit parameters as defined in the official
     KM3NeT-Dataformat.
@@ -56,6 +57,7 @@ def fitparams():
     """
     return fitparameters.keys()
 
+
 def count_nested(Array, axis=0):
     """count elements in a nested awkward Array.
 
@@ -80,6 +82,7 @@ def count_nested(Array, axis=0):
     if axis == 2:
         return ak1.count(Array, axis=2)
 
+
 def best_track(tracks, strategy="first", rec_stages=None):
     """best track selection based on different strategies
 
@@ -96,7 +99,6 @@ def best_track(tracks, strategy="first", rec_stages=None):
     #     mask = tracks.rec_stages[]
 
 
-
 EVENTS_MAP = BranchMapper(name="events",
                           key="Evt",
                           extra={

tests/test_offline.py

@@ -36,6 +36,7 @@ class TestFitinf(unittest.TestCase):
 
         assert "JGANDALF_BETA0_RAD" in keys
 
+
 class TestCountNested(unittest.TestCase):
     def test_count_nested(self):
         fit = OFFLINE_FILE.events.tracks.fitinf