Skip to content

Assigned idx_remain if self.drop == False to run w/ permE aug #258

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

Open
wants to merge 1 commit into
base: dig-stable
Choose a base branch
from

Conversation

jonathanrppittman
Copy link

@jonathanrppittman jonathanrppittman commented Jul 1, 2025

The do_trans() function in /sslgraph/method/contrastive/views_fn/structure.py previously used the idx_remain variable before it was assigned to anything if the class variable, drop, was False, causing an error. This change fixes this by assigning idx_remain to data.edge_index when self.drop is False. Thus, this code will now run whether self.drop is True or False.

@jonathanrppittman jonathanrppittman changed the title Assigned idx_remain as if self.drop == False to run w/ permE aug Assigned idx_remain if self.drop == False to run w/ permE aug Jul 1, 2025
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