Merge branch 'master' into feat/skorch-compatible

Author: 34j

.pre-commit-config.yaml

@@ -27,7 +27,7 @@ repos:
           )$
 
   - repo: https://github.com/adrienverge/yamllint.git
-    rev: v1.33.0
+    rev: v1.35.1
     hooks:
       - id: yamllint
         name: Lint yaml
@@ -47,21 +47,21 @@ repos:
         name: Sort imports
 
   - repo: https://github.com/PyCQA/flake8
-    rev: 6.1.0
+    rev: 7.0.0
     hooks:
       - id: flake8
         name: Check PEP8
         additional_dependencies: [Flake8-pyproject]
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.1.9
+    rev: v0.3.5
     hooks:
       - id: ruff
         name: Ruff formatting
         args: [--fix, --exit-non-zero-on-fix]
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.8.0
+    rev: v1.9.0
     hooks:
       - id: mypy
         name: Check types

CITATION.cff

@@ -5,8 +5,6 @@ title: "PyTorch Frame: A Deep Learning Framework for Tabular Data"
 authors:
 - family-names: "Hu"
   given-names: "Weihua"
-- family-names: "Fey"
-  given-names: "Matthias"
 - family-names: "Yuan"
   given-names: "Yiwen"
 - family-names: "Zhang"
@@ -17,6 +15,10 @@ authors:
   given-names: "Kaidi"
 - family-names: "Kocijan"
   given-names: "Vid"
+- family-names: "Leskovec"
+  given-names: "Jure"
+- family-names: "Fey"
+  given-names: "Matthias"
 date-released: 2023-10-24
 license: MIT
 url: "https://github.com/pyg-team/pytorch-frame"

README.md

@@ -264,3 +264,15 @@ pip install pytorch_frame
 ```
 
 See [the installation guide](https://pytorch-frame.readthedocs.io/en/latest/get_started/installation.html) for other options.
+
+## Cite
+
+If you use PyTorch Frame in your work, please cite our paper (Bibtex below).
+```
+@article{hu2024pytorch,
+  title={PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning},
+  author={Hu, Weihua and Yuan, Yiwen and Zhang, Zecheng and Nitta, Akihiro and Cao, Kaidi and Kocijan, Vid and Leskovec, Jure and Fey, Matthias},
+  journal={arXiv preprint arXiv:2404.00776},
+  year={2024}
+}
+```

pyproject.toml

@@ -50,6 +50,7 @@ full=[
     "scikit-learn",
     "xgboost>=1.7.0, <2.0.0",
     "optuna>=3.0.0",
+    "optuna-integration",
     "mpmath==1.3.0",
     "catboost",
     "lightgbm",

torch_frame/testing/image_embedder.py

@@ -22,5 +22,5 @@ def __init__(
         super().__init__()
         self.out_channels = out_channels
 
-    def forward_embed(self, images: list[Image]) -> Tensor:
+    def forward_embed(self, images: list[Image.Image]) -> Tensor:
         return torch.rand(len(images), self.out_channels)