diff --git a/src/signatory/__init__.py b/src/signatory/__init__.py index ea3d740..6dbcd3e 100644 --- a/src/signatory/__init__.py +++ b/src/signatory/__init__.py @@ -20,6 +20,7 @@ import torch # must be imported before anything from signatory +import sklearn try: from . import impl