Fixing some more imports in SFNO examples

torch_harmonics/examples/sfno/models/layers.py

@@ -29,21 +29,11 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #
 
-from functools import partial
-from collections import OrderedDict
-from copy import Error, deepcopy
-from re import S
-from numpy.lib.arraypad import pad
-import numpy as np
-
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
 import torch.fft
-from torch.nn.modules.container import Sequential
-from torch.utils.checkpoint import checkpoint, checkpoint_sequential
+from torch.utils.checkpoint import checkpoint
 from torch.cuda import amp
-from typing import Optional
 import math
 
 from torch_harmonics import *

torch_harmonics/examples/sfno/utils/pde_dataset.py

@@ -30,7 +30,6 @@
 #
 
 import torch
-import os
 
 from math import ceil