Merge branch 'master' into feat/skorch-compatible

pyg-team · Sep 23, 2024 · eddecf8 · eddecf8
2 parents 8e46fb5 + cae7ccf
commit eddecf8
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/documentation.yml b/.github/workflows/documentation.yml
@@ -23,7 +23,6 @@ jobs:
         uses: tj-actions/changed-files@v41
         with:
           files: |
-            docs/**
             examples/**
             README.md
             CHANGELOG.md

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -40,7 +40,7 @@
     'python': ('https://docs.python.org/3', None),
     'numpy': ('http://docs.scipy.org/doc/numpy', None),
     'pandas': ('http://pandas.pydata.org/pandas-docs/dev', None),
-    'torch': ('https://pytorch.org/docs/master', None),
+    'torch': ('https://pytorch.org/docs/stable', None),
     'optuna': ('https://optuna.readthedocs.io/en/stable/', None),
     'xgboost': ('https://xgboost.readthedocs.io/en/stable/', None),
 }

diff --git a/torch_frame/nn/encoder/stype_encoder.py b/torch_frame/nn/encoder/stype_encoder.py
@@ -732,6 +732,7 @@ def encode_forward(
             # -> [batch_size, out_channels]
             x_lin = feat.values[:, start_idx:end_idx] @ self.weight_list[idx]
             x_lins.append(x_lin)
+            start_idx = end_idx
         # [batch_size, num_cols, out_channels]
         x = torch.stack(x_lins, dim=1)
         # [batch_size, num_cols, out_channels] + [num_cols, out_channels]