Commit 82541d0f authored by nd-02110114's avatar nd-02110114
Browse files

🚨 fix lint

parent e27b788f
Loading
Loading
Loading
Loading
+1 −2
Original line number Original line Diff line number Diff line
@@ -95,8 +95,7 @@ class MoleculeGraphData(object):
      import torch
      import torch
      from torch_geometric.data import Data
      from torch_geometric.data import Data
    except ModuleNotFoundError:
    except ModuleNotFoundError:
      raise ValueError(
      raise ValueError("This class requires PyTorch Geometric to be installed.")
          "This class requires PyTorch Geometric to be installed.")


    return Data(
    return Data(
      x=torch.from_numpy(self.node_features),
      x=torch.from_numpy(self.node_features),