Update stafnet.py

dylan-yin · web-flow · commit 7bb6fc726f87 · 2025-06-23T20:51:35.000+08:00
diff --git a/ppsci/arch/stafnet.py b/ppsci/arch/stafnet.py
@@ -6,7 +6,7 @@
 import paddle
 from ppsci.arch import base
 try:
-    import pgl
+    from pgl.nn.conv import GATv2Conv
 except ModuleNotFoundError:
     pass
 
