Skip to content

Commit

Permalink
Fix Lasso interop issue (#5116)
Browse files Browse the repository at this point in the history
Answers #5106

Authors:
  - Victor Lafargue (https://github.com/viclafargue)

Approvers:
  - Dante Gama Dessavre (https://github.com/dantegd)

URL: #5116
  • Loading branch information
viclafargue committed Jan 5, 2023
1 parent 0303a93 commit 62a2661
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions python/cuml/internals/api_decorators.py
Expand Up @@ -785,6 +785,10 @@ def device_interop_preparation(init_func):

@functools.wraps(init_func)
def processor(self, *args, **kwargs):
# if child class is already prepared for interop, skip
if hasattr(self, '_full_kwargs'):
return init_func(self, *args, **kwargs)

# Save all kwargs
self._full_kwargs = kwargs
# Generate list of available cuML hyperparameters
Expand Down

0 comments on commit 62a2661

Please sign in to comment.