Update README with author information and paper details; remove obsol…

…ete test file
uhh-pd-ml · Jan 13, 2025 · 9f01b6c · 9f01b6c
1 parent d4b55ea
commit 9f01b6c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 55 deletions.
diff --git a/README.md b/README.md
@@ -1,7 +1,16 @@
 # OmniJet-α Calo: Applying OJA to calorimeter data
 
-This repository contains the code for the results presented in the paper `link`
-The documentation of the original OmniJet-α can be found at [joschkabirk.github.io/gabbro](https://joschkabirk.github.io/gabbro/)
+<div style="text-align: center;"> 
+Joschka Birk, Frank Gaede, Anna Hallin, Gregor Kasieczka, Martina Mozzanica, Henning Rose
+
+[![arXiv](https://img.shields.io/badge/arXiv-2501.05534-red)](https://arxiv.org/abs/2501.05534)
+[![PyTorch](https://img.shields.io/badge/PyTorch-2.2-orange)](https://pytorch.org)
+[![Lightning](https://img.shields.io/badge/Lightning-2.2.1-purple)](https://lightning.ai)
+[![Hydra](https://img.shields.io/badge/Hydra-1.3-blue)](https://hydra.cc)
+</div>
+
+This repository contains the code for the results presented in the paper [`OmniJet-α_C: Learning point cloud calorimeter simulations using generative transformers`](https://arxiv.org/abs/2501.05534)
+The documentation of the original OmniJet-α can be found at [uhh-pd-ml/omnijet_alpha](https://github.com/uhh-pd-ml/omnijet_alpha)
 
 **Abstract:**
 

diff --git a/tests/test_utils_jet_types.py b/tests/test_utils_jet_types.py