Skip to content

Commit

Permalink
typing stuff
Browse files Browse the repository at this point in the history
  • Loading branch information
DavidLapous committed Sep 2, 2024
1 parent 1aed399 commit 921c500
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions multipers/ml/convolutions.py
Original file line number Diff line number Diff line change
Expand Up @@ -349,7 +349,7 @@ def score_samples(self, Y, X=None, return_kernel=False):
log_probs : tensor (m)
log probability densities for each of the queried points in `Y`
"""
assert self._backend is not None, "Fit first."
assert self._backend is not None and self._kernel is not None, "Fit first."
X = self.X if X is None else X
if X.shape[0] == 0:
return self._backend.zeros((Y.shape[0]))
Expand Down Expand Up @@ -418,7 +418,7 @@ class DTM:
Distance To Measure
"""

def __init__(self, masses=[], metric: str = "euclidean", **_kdtree_kwargs):
def __init__(self, masses, metric: str = "euclidean", **_kdtree_kwargs):
"""
mass : float in [0,1]
The mass threshold
Expand Down
2 changes: 1 addition & 1 deletion multipers/ml/mma.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class FilteredComplex2MMA(BaseEstimator, TransformerMixin):

def __init__(
self,
n_jobs: int = 1,
n_jobs: int = -1,
expand_dim: Optional[int] = None,
prune_degrees_above: Optional[int] = None,
progress=False,
Expand Down

0 comments on commit 921c500

Please sign in to comment.