From 4542d745f168fba458301bc9926a843880d5f8b0 Mon Sep 17 00:00:00 2001
From: sovietdevil <2558390548@qq.com>
Date: Fri, 26 Jul 2024 15:54:52 +0200
Subject: [PATCH] Add ShiftML version 2 example, and correct the variable name
 in readme

---
 README.md                     | 2 +-
 src/shiftml/ase/calculator.py | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index fd60cec..3ad8126 100644
--- a/README.md
+++ b/README.md
@@ -21,7 +21,7 @@ from shiftml.ase import ShiftML
 frame = bulk("C", "diamond", a=3.566)
 calculator = ShiftML("ShiftML1.0")
 
-cs_iso = calc.get_cs_iso(frame)
+cs_iso = calculator.get_cs_iso(frame)
 
 print(cs_iso)
 
diff --git a/src/shiftml/ase/calculator.py b/src/shiftml/ase/calculator.py
index 791f223..c2ad35d 100644
--- a/src/shiftml/ase/calculator.py
+++ b/src/shiftml/ase/calculator.py
@@ -14,16 +14,19 @@
 url_resolve = {
     "ShiftML1.0": "https://tinyurl.com/3xwec68f",
     "ShiftML1.1": "https://tinyurl.com/53ymkhvd",
+    "ShiftML2.0": "https://tinyurl.com/9v8ppnru",
 }
 
 resolve_outputs = {
     "ShiftML1.0": {"mtt::cs_iso": ModelOutput(quantity="", unit="ppm", per_atom=True)},
     "ShiftML1.1": {"mtt::cs_iso": ModelOutput(quantity="", unit="ppm", per_atom=True)},
+    "ShiftML2.0": {"mtt::cs_iso": ModelOutput(quantity="", unit="ppm", per_atom=True)},
 }
 
 resolve_fitted_species = {
     "ShiftML1.0": set([1, 6, 7, 8, 16]),
     "ShiftML1.1": set([1, 6, 7, 8, 16]),
+    "ShiftML2.0": set([1, 6, 7, 8, 9, 11, 12, 15, 16, 17, 19, 20]),
 }