Skip to content

Commit 93a641d

Browse files
committed
Fix to model catalog to use sklearn model by default
1 parent b2cbfad commit 93a641d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

cyclops/models/catalog.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,12 +186,12 @@ def wrap_model(model: Union[Module, BaseEstimator], **kwargs) -> WrappedModel:
186186
If model is not a pyTorch or sklearn model.
187187
188188
"""
189+
if is_sklearn_model(model):
190+
return SKModel(model, **kwargs)
189191
if is_pytorch_model(model):
190192
from cyclops.models.wrappers import PTModel
191193

192194
return PTModel(model, **kwargs)
193-
if is_sklearn_model(model):
194-
return SKModel(model, **kwargs)
195195
raise TypeError("``model`` must be a PyTorch or sklearn model")
196196

197197

0 commit comments

Comments
 (0)