Merge 03434a7 into 08a45b7

geoopt/manifolds/base.py

@@ -1,6 +1,6 @@
 from collections import defaultdict
 import abc
-import torch
+import torch.nn
 import re
 
 __all__ = ["Manifold"]
@@ -148,7 +148,7 @@ def not_implemented(*args, **kwargs):
     raise NotImplementedError
 
 
-class Manifold(metaclass=ManifoldMeta):
+class Manifold(torch.nn.Module, metaclass=ManifoldMeta):
     r"""
     Base class for Manifolds
 
@@ -191,6 +191,13 @@ class Manifold(metaclass=ManifoldMeta):
     reversible = None
     _default_order = 1
 
+    def __init__(self, **kwargs):
+        super().__init__()
+
+    def forward(self, *input):
+        # this removes all warnings about implementing abstract methods
+        raise TypeError("Manifold is not callable")
+
     # noinspection PyAttributeOutsideInit
     def set_default_order(self, order):
         """

geoopt/manifolds/sphere.py

@@ -116,6 +116,7 @@ class SphereSubspaceIntersection(Sphere):
     name = "SphereSubspace"
 
     def __init__(self, span):
+        super().__init__()
         self._configure_manifold(span)
         if (geoopt.linalg.batch_linalg.matrix_rank(self._projector) == 1).any():
             raise ValueError(