version up

README.md

@@ -1,6 +1,6 @@
 ![](docs/pygda_logo.png)
 -----
-[![PyPI version](https://badge.fury.io/py/pygda.svg)](https://badge.fury.io/py/pygda)
+[![PyPI - Version](https://img.shields.io/pypi/v/pygda?style=flat)](https://pypi.org/project/pygda/)
 [![Documentation Status](https://readthedocs.org/projects/pygda/badge/?version=stable)](https://pygda.readthedocs.io/en/stable/?badge=stable)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 [![contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat)](https://github.com/pygda-team/pygda/issues)

pygda/models/strurw.py

@@ -234,7 +234,7 @@ def forward_model_mixup(self, source_data, target_data, epoch):
             data_b.edge_index,
             lam,
             id_new_value_old,
-            edge_weight)
+            source_data.edge_weight)
         source_logits = self.gnn.feat_classifier(source_feat)
 
         loss = F.nll_loss(F.log_softmax(source_logits, dim=1), source_data.y)

pygda/nn/reweight_gnn.py

@@ -113,7 +113,6 @@ def __init__(
         **kwargs
         ):
 
-        # kwargs.setdefault('aggr', "add")
         super(GCN_reweight, self).__init__(aggr=aggr, flow ="target_to_source")
 
         self.in_channels = in_channels

pygda/version.py

@@ -1 +1 @@
-__version__ = '0.0.4'
+__version__ = '0.0.5'