Skip to content

Bug Fix: Assert only fixed_atom_mask and fixed_edge_mask can be None in to_xace_mol#61

Open
juhimgupta wants to merge 4 commits intodev_gnnfrom
protein_noise_feat
Open

Bug Fix: Assert only fixed_atom_mask and fixed_edge_mask can be None in to_xace_mol#61
juhimgupta wants to merge 4 commits intodev_gnnfrom
protein_noise_feat

Conversation

@juhimgupta
Copy link
Collaborator

File changed: omtra/data/plinder/init.py

  • Replaced try/except block in to_xace_mol with explicit assertion
  • If xace_dict[k] is None, asserts that k in ['fixed_atom_mask', 'fixed_edge_mask']
  • Raises AssertionError with message f"{k} is None but not in ['fixed_atom_mask', 'fixed_edge_mask']
  • Skips .long() conversion for None values

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant