Skip to content

Commit

Permalink
add import warning to visnet import guard
Browse files Browse the repository at this point in the history
  • Loading branch information
fyng committed Feb 5, 2024
1 parent 34c8ace commit ba91938
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mtenn/conversion_utils/visnet.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Representation and strategy for ViSNet model.
"""
import warning
from copy import deepcopy
import torch
from torch.autograd import grad
Expand All @@ -16,8 +17,7 @@
from torch_geometric.nn.models import ViSNet as PygVisNet
HAS_VISNET = True
except ImportError:
pass

warning.warn("VisNet import error. Is your PyG >=2.5.0? Refer to issue #42", ImportWarning)

class EquivariantVecToScalar(torch.nn.Module):
# Wrapper for PygVisNet.EquivariantScalar to implement forward() method
Expand Down

0 comments on commit ba91938

Please sign in to comment.