Skip to content

Commit

Permalink
run blackk
Browse files Browse the repository at this point in the history
  • Loading branch information
samwaseda committed Aug 5, 2024
1 parent 94e89af commit a4ee121
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions elaston/green.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,9 +58,7 @@ def _get_greens_function(self, r, derivative=0, fourier=False):
"get_greens_function must be defined in the child class"
)

def get_greens_function(
self, r, derivative=0, fourier=False, check_unique=False
):
def get_greens_function(self, r, derivative=0, fourier=False, check_unique=False):
"""
Args:
r ((n,3)-array): Positions for which to calculate the Green's function
Expand All @@ -77,9 +75,7 @@ def get_greens_function(
x = np.array(r)
if check_unique:
x, inv = np.unique(x.reshape(-1, 3), axis=0, return_inverse=True)
g_tmp = self._get_greens_function(
r=x, derivative=derivative, fourier=fourier
)
g_tmp = self._get_greens_function(r=x, derivative=derivative, fourier=fourier)
if check_unique:
g_tmp = g_tmp[inv].reshape(np.asarray(r).shape + (derivative + 1) * (3,))
return g_tmp
Expand Down

0 comments on commit a4ee121

Please sign in to comment.