adding spherical attention

torch_harmonics/examples/models/__init__.py

@@ -29,5 +29,8 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #
 
-from .sfno import SphericalFourierNeuralOperatorNet
-from .lsno import LocalSphericalNeuralOperatorNet
+from .sfno import SphericalFourierNeuralOperator
+from .lsno import LocalSphericalNeuralOperator
+from .s2unet import SphericalUNet
+from .s2transformer import SphericalTransformer
+from .s2segformer import SphericalSegformer