replace import from torch_cfd.equations with torch_cfd.spectral

scaomath · scaomath · commit 4fc7924adc93 · 2025-05-13T16:17:28.000-05:00
diff --git a/fno/sfno_pytest.py b/fno/sfno_pytest.py
@@ -11,7 +11,7 @@
     SpectralConvS,
     SpectralConvT,
 )
-from torch_cfd.equations import *
+from torch_cfd.spectral import *
 from contextlib import contextmanager
 
 

Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,7 @@`
`11`	`11`	`SpectralConvS,`
`12`	`12`	`SpectralConvT,`
`13`	`13`	`)`
`14`		`-from torch_cfd.equations import *`
	`14`	`+from torch_cfd.spectral import *`
`15`	`15`	`from contextlib import contextmanager`
`16`	`16`
`17`	`17`