Skip to content

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