Minor formatting

Eden-Kramer-Lab · Dec 11, 2017 · aa2e8c5 · aa2e8c5
1 parent 28abbf5
commit aa2e8c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/replay_classification/decoders.py b/replay_classification/decoders.py
@@ -467,7 +467,8 @@ def __init__(self, results, spikes=None):
         self.spikes = spikes
 
     def state_probability(self):
-        return self.results['posterior_density'].sum('position').to_series().unstack()
+        return (self.results['posterior_density'].sum('position')
+                .to_series().unstack())
 
     def predicted_state(self):
         state_probability = self.state_probability()