From 5362f4901c0b5a952df512faabc0a5235c09adaf Mon Sep 17 00:00:00 2001
From: Weihua Hu <weihua916@gmail.com>
Date: Tue, 2 Apr 2024 09:41:17 -0700
Subject: [PATCH 1/4] update

---
 README.md | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/README.md b/README.md
index a7af6a10..39a7667b 100644
--- a/README.md
+++ b/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}
+}
+```

From acf3b48435d7f0c3ad0ced3679e14e3693529853 Mon Sep 17 00:00:00 2001
From: Weihua Hu <weihua916@gmail.com>
Date: Tue, 2 Apr 2024 09:56:15 -0700
Subject: [PATCH 2/4] update

---
 torch_frame/testing/image_embedder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_frame/testing/image_embedder.py b/torch_frame/testing/image_embedder.py
index b52b13ac..f6689220 100644
--- a/torch_frame/testing/image_embedder.py
+++ b/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)

From 04befebc994f16674cdaca96e1a2352d2856b6f8 Mon Sep 17 00:00:00 2001
From: Weihua Hu <weihua916@gmail.com>
Date: Tue, 2 Apr 2024 10:18:57 -0700
Subject: [PATCH 3/4] update

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 80069e28..7f8d44d3 100644
--- a/pyproject.toml
+++ b/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",

From 5bdd0dad6a9ffe030f854e21d84e587df6b589e0 Mon Sep 17 00:00:00 2001
From: Weihua Hu <weihua916@gmail.com>
Date: Tue, 2 Apr 2024 10:55:47 -0700
Subject: [PATCH 4/4] update

---
 CITATION.cff | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/CITATION.cff b/CITATION.cff
index 5528e85a..24da427f 100644
--- a/CITATION.cff
+++ b/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"