add zenodo doi

RMeli · Jul 30, 2022 · 1c91777 · 1c91777
1 parent c26cdc8
commit 1c91777
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -4,6 +4,8 @@
 [![GitHub Actions Build Status](https://github.com/RMeli/gnina-torch/workflows/CI/badge.svg)](https://github.com/RMeli/gnina-torch/actions?query=workflow%3ACI)
 [![codecov](https://codecov.io/gh/RMeli/gnina-torch/branch/main/graph/badge.svg?token=KjVkShwQ1z)](https://codecov.io/gh/RMeli/gnina-torch)
 
+[![DOI](https://zenodo.org/badge/419005477.svg)](https://zenodo.org/badge/latestdoi/419005477)
+
 PyTorch implementation of [GNINA](https://github.com/gnina/gnina) scoring function.
 
 ## References
@@ -75,7 +77,7 @@ Pre-trained GNINA models can be loaded as follows:
 ```python
 from gninatorch.gnina import setup_gnina_model
 
-model = setup_gnina_model(MODEL)```
+model = setup_gnina_model(MODEL)
 ```
 where `MODEL` corresponds to the `--cnn` argument in [GNINA](https://github.com/gnina/gnina).
 

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 import versioneer
 
-short_description = "PyTorch implementation of GNINA".split("\n")[0]
+short_description = "PyTorch implementation of GNINA scoring function".split("\n")[0]
 
 # from https://github.com/pytest-dev/pytest-runner#conditional-requirement
 needs_pytest = {"pytest", "test", "ptr"}.intersection(sys.argv)