Merge pull request #4813 from FederatedAI/feature-1.11.1-doc-update

fix import when geometric is not installed

Author: mgqa34

python/federatedml/nn/dataset/graph.py

@@ -2,7 +2,10 @@
 import pandas as pd
 from federatedml.statistic.data_overview import with_weight
 from federatedml.nn.dataset.base import Dataset
-from torch_geometric.data import Data
+try:
+    from torch_geometric.data import Data
+except BaseException:
+    pass
 import torch
 from federatedml.util import LOGGER