Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Incorrect Value Replacement in edge_types #12

Open
FilippNikitin opened this issue Dec 17, 2024 · 0 comments
Open

Incorrect Value Replacement in edge_types #12

FilippNikitin opened this issue Dec 17, 2024 · 0 comments

Comments

@FilippNikitin
Copy link

Description

The line:

edge_types[edge_types == 4] = 1.5

fails because edge_types is a long tensor, truncating 1.5 to 1. This replaces 4 with 1 instead of 1.5, affecting logic and results.

Impact:
This propagates incorrect values, impacting downstream research like molecule and atom stabilities in works such as EQGAT-Diff and Semla-Flow.

Suggested Fix:

Convert edge_types to a floating-point tensor before replacement:

edge_types = edge_types.float()
edge_types[edge_types == 4] = 1.5

edge_types[edge_types == 4] = 1.5

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

No branches or pull requests

1 participant