Skip to content

Commit 803eb28

Browse files
authored
Update stafnet_dataset.py
1 parent 3e35a0d commit 803eb28

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ppsci/data/dataset/stafnet_dataset.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,11 @@
44
import numpy as np
55
import paddle
66
import pandas
7-
import pgl
7+
try:
8+
import pgl
9+
except ModuleNotFoundError as e:
10+
logger.error("please install pgl with \'pip install pgl\'.")
11+
raise ModuleNotFoundError(str(e))
812
from paddle import io
913
from paddle.io import DataLoader
1014
from scipy.spatial.distance import cdist

0 commit comments

Comments
 (0)