Skip to content

Commit

Permalink
STY : Correction of syntax
Browse files Browse the repository at this point in the history
Correct for the syntax issues suggested by the PEP8 bot nipreps#50
  • Loading branch information
celprov committed Apr 26, 2021
1 parent 6aa9ff5 commit 6caff36
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion eddymotion/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,6 +90,7 @@ def predict(self, gradient, **kwargs):
"""Return the *b=0* map."""
return self._S0


class AverageDWModel:
"""A trivial model that returns an average map."""

Expand All @@ -101,12 +102,13 @@ def __init__(self, gtab, **kwargs):

def fit(self, data, **kwargs):
"""Calculate the average."""
self._data = data.mean(-1)
self._data = data.mean(-1)

def predict(self, gradient, **kwargs):
"""Return the average map."""
return self._data


class DTIModel:
"""A wrapper of :obj:`dipy.reconst.dti.TensorModel."""

Expand Down

0 comments on commit 6caff36

Please sign in to comment.