Add citation

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)