Skip to content

Commit

Permalink
Style fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
dantegd committed Oct 8, 2024
1 parent 1086f75 commit abf81bd
Show file tree
Hide file tree
Showing 6 changed files with 12 additions and 8 deletions.
4 changes: 2 additions & 2 deletions python/cuml/cuml/dask/cluster/dbscan.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2020-2024, NVIDIA CORPORATION.
# Copyright (c) 2020-2023, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -161,4 +161,4 @@ def fit_predict(self, X, out_dtype="int32"):
return self.get_combined_model().labels_

def get_param_names(self):
return list(cls.kwargs.keys())
return list(self.kwargs.keys())
2 changes: 1 addition & 1 deletion python/cuml/cuml/dask/decomposition/pca.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2019-2024, NVIDIA CORPORATION.
# Copyright (c) 2019-2023, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down
2 changes: 1 addition & 1 deletion python/cuml/cuml/dask/decomposition/tsvd.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2019-2024, NVIDIA CORPORATION.
# Copyright (c) 2019-2023, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down
2 changes: 1 addition & 1 deletion python/cuml/cuml/dask/linear_model/linear_regression.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2019-2024, NVIDIA CORPORATION.
# Copyright (c) 2019-2023, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down
2 changes: 1 addition & 1 deletion python/cuml/cuml/dask/linear_model/ridge.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2019-2024, NVIDIA CORPORATION.
# Copyright (c) 2019-2023, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down
8 changes: 6 additions & 2 deletions python/cuml/cuml/decomposition/incremental_pca.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,7 +216,6 @@ def __init__(
output_type=output_type,
)
self.batch_size = batch_size
self._hyperparams = ["n_components", "whiten", "copy", "batch_size"]
self._sparse_model = True

def fit(self, X, y=None, convert_dtype=True) -> "IncrementalPCA":
Expand Down Expand Up @@ -452,7 +451,12 @@ def transform(self, X, convert_dtype=False) -> CumlArray:
@classmethod
def get_param_names(cls):
# Skip super() since we dont pass any extra parameters in __init__
return Base.get_param_names() + self._hyperparams
return Base.get_param_names() + [
"n_components",
"whiten",
"copy",
"batch_size",
]


def _validate_sparse_input(X):
Expand Down

0 comments on commit abf81bd

Please sign in to comment.