stjames 0.0.90__py3-none-any.whl → 0.0.92__py3-none-any.whl

stjames/engine.py

@@ -0,0 +1,16 @@
+from .base import LowercaseStrEnum
+
+
+class Mode(LowercaseStrEnum):
+    AIMNET2 = "aimnet2"
+    MACE = "mace"
+    OCP24 = "ocp24"
+    OMOL25 = "omol25"
+    ORB = "orb"
+    TBLITE = "tblite"
+    XTB = "xtb"
+    TERRACHEM = "terrachem"
+    PYSCF = "pyscf"
+    PSI4 = "psi4"
+    OPENFF = "openff"
+    EGRET = "egret"

stjames/method.py

@@ -37,7 +37,7 @@ class Method(LowercaseStrEnum):
     OCP24_L = "ocp24_l"
     OMOL25_CONSERVING_S = "omol25_conserving_s"
     UMA_M_OMOL = "uma_m_omol"
-    ORB_V2 = "orb_v2"
+    UMA_S_OMOL = "uma_s_omol"
     ORB_V3_CONSERVATIVE_INF_OMAT = "orb_v3_conservative_inf_omat"
 
     GFN_FF = "gfn_ff"
@@ -51,12 +51,53 @@ class Method(LowercaseStrEnum):
 
     OFF_SAGE_2_2_1 = "off_sage_2_2_1"
 
-    RM1 = "rm1"
-
     EGRET_1 = "egret_1"
     EGRET_1E = "egret_1e"
     EGRET_1T = "egret_1t"
 
+    def default_engine(self, *, is_periodic: bool = False) -> str:
+        """
+        Return the canonical engine name for this quantum-chemistry method.
+
+        :param bool is_periodic:
+            If ``True`` **and** the method is in the XTB family, return
+            ``"tblite"`` (periodic-capable backend) instead of ``"xtb"``.
+        :returns: Lower-case engine identifier (e.g. ``"psi4"``, ``"mace"``).
+
+        **Examples**
+
+        .. code-block:: python
+
+           Method.MACE_MP_0B2_L.default_engine()
+           # 'mace'
+
+           Method.GFN2_XTB.default_engine()
+           # 'xtb'
+
+           Method.GFN2_XTB.default_engine(is_periodic=True)
+           # 'tblite'
+        """
+        match self:
+            case Method.AIMNET2_WB97MD3:
+                return "aimnet2"
+            case Method.MACE_MP_0B2_L:
+                return "mace"
+            case Method.OCP24_S | Method.OCP24_L:
+                return "ocp24"
+            case Method.OMOL25_CONSERVING_S | Method.UMA_M_OMOL | Method.UMA_S_OMOL:
+                return "omol25"
+            case Method.ORB_V3_CONSERVATIVE_INF_OMAT:
+                return "orb"
+            case _ if self in XTB_METHODS:
+                return "tblite" if is_periodic else "xtb"
+            case Method.OFF_SAGE_2_2_1:
+                return "openff"
+            case Method.EGRET_1 | Method.EGRET_1E | Method.EGRET_1T:
+                return "egret"
+            case _:
+                # All remaining methods (HF, DFT, composite, etc.) fall back to Psi4
+                return "psi4"
+
 
 PrepackagedNNPMethod = Literal[
     Method.AIMNET2_WB97MD3,
@@ -64,7 +105,7 @@ PrepackagedNNPMethod = Literal[
     Method.OCP24_L,
     Method.OMOL25_CONSERVING_S,
     Method.UMA_M_OMOL,
-    Method.RM1,
+    Method.UMA_S_OMOL,
     Method.ORB_V3_CONSERVATIVE_INF_OMAT,
     Method.EGRET_1,
     Method.EGRET_1E,
@@ -77,15 +118,15 @@ PREPACKAGED_NNP_METHODS = [
     Method.OCP24_L,
     Method.OMOL25_CONSERVING_S,
     Method.UMA_M_OMOL,
-    Method.RM1,
+    Method.UMA_S_OMOL,
     Method.ORB_V3_CONSERVATIVE_INF_OMAT,
     Method.EGRET_1,
     Method.EGRET_1E,
     Method.EGRET_1T,
 ]
 
-CorrectableNNPMethod = Literal[Method.MACE_MP_0B2_L, Method.ORB_V2]
-CORRECTABLE_NNP_METHODS = [Method.MACE_MP_0B2_L, Method.ORB_V2]
+CorrectableNNPMethod = Literal[Method.MACE_MP_0B2_L]
+CORRECTABLE_NNP_METHODS = [Method.MACE_MP_0B2_L]
 
 NNPMethod = PrepackagedNNPMethod | CorrectableNNPMethod
 NNP_METHODS = [*PREPACKAGED_NNP_METHODS, *CORRECTABLE_NNP_METHODS]

{stjames-0.0.90.dist-info → stjames-0.0.92.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stjames
-Version: 0.0.90
+Version: 0.0.92
 Summary: standardized JSON atom/molecule encoding scheme
 Author-email: Corin Wagen <corin@rowansci.com>
 Project-URL: Homepage, https://github.com/rowansci/stjames

{stjames-0.0.90.dist-info → stjames-0.0.92.dist-info}/RECORD

@@ -8,10 +8,11 @@ stjames/compute_settings.py,sha256=wuYE6W4WqP3oFHwAeEp7XQ0oMz_MNLucWgrsuJ4kEFQ,2
 stjames/constraint.py,sha256=B6oV0rYjmAWr8gpi5f03gRy_uuqjUURVDVwoez5Cfbg,2442
 stjames/correction.py,sha256=ZVErCcj4TPyZeKrdvXVjHa0tFynsCaoy96QZUVxWFM8,413
 stjames/diis_settings.py,sha256=4m1EQQWBlpHhMnWopix8qOqJv7QCluvdnV9jSKJDFtE,552
+stjames/engine.py,sha256=wVItZQSIeR3ap6EZG3kOw0xCZztZm-6KKFrX84ZyFsE,313
 stjames/grid_settings.py,sha256=WrSNGc-8_f87YBZYt9Hh7RbhM4MweADoVzwBMcSqcsE,640
 stjames/int_settings.py,sha256=5HXp8opt5ZyY1UpmfaK7NVloWVLM5jkG0elEEqpVLUo,896
 stjames/message.py,sha256=Rq6QqmHZKecWxYH8fVyXmuoCCPZv8YinvgykSeorXSU,216
-stjames/method.py,sha256=aPFUhfUtJ_befRHO1Aa_c4ofEQAPZlQUvHWLaCv32Z4,2945
+stjames/method.py,sha256=8Cj0JYGMEaS_Cetfz8Wcu6-BZAv1SejRSagMgkKaXs0,4469
 stjames/mode.py,sha256=xw46Cc7f3eTS8i35qECi-8DocAlANhayK3w4akD4HBU,496
 stjames/molecule.py,sha256=l4S6prH1xflnZtbwidSF2THsiHaSJtRiy1rmUbH366Q,20631
 stjames/opt_settings.py,sha256=LEwGXUEKq5TfU5rr60Z4QQBhCqiw1Ch5w0M_lXawWo8,642
@@ -62,8 +63,8 @@ stjames/workflows/solubility.py,sha256=kGfVyPPGDLRpf2j6dSY7woCkfsoXSbUzdSImA4mcM
 stjames/workflows/spin_states.py,sha256=0degmE-frovgoXweshZyjfjqL7nkbaFoO9YoJhvQnaI,4748
 stjames/workflows/tautomer.py,sha256=7eYKziGPg8Km6lfowTzSkgJfJ4SHUPrAmnTf8Bi-SB0,1164
 stjames/workflows/workflow.py,sha256=OE05pt2ZOd8TzTOlBngXCVg9wv_553ZR60VNRPlq0f8,1953
-stjames-0.0.90.dist-info/licenses/LICENSE,sha256=i7ehYBS-6gGmbTcgU4mgk28pyOx2kScJ0kcx8n7bWLM,1084
-stjames-0.0.90.dist-info/METADATA,sha256=IbwiPJaGd1rkTp0eSNDf5yTAHqnJ21wTIgI18ObeaHA,1724
-stjames-0.0.90.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-stjames-0.0.90.dist-info/top_level.txt,sha256=FYCwxl6quhYOAgG-mnPQcCK8vsVM7B8rIUrO-WrQ_PI,8
-stjames-0.0.90.dist-info/RECORD,,
+stjames-0.0.92.dist-info/licenses/LICENSE,sha256=i7ehYBS-6gGmbTcgU4mgk28pyOx2kScJ0kcx8n7bWLM,1084
+stjames-0.0.92.dist-info/METADATA,sha256=QQ4yZAEcRFwv4KCTOBZxuCrOzmNb_kBv2G3Jd4-2hR4,1724
+stjames-0.0.92.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+stjames-0.0.92.dist-info/top_level.txt,sha256=FYCwxl6quhYOAgG-mnPQcCK8vsVM7B8rIUrO-WrQ_PI,8
+stjames-0.0.92.dist-info/RECORD,,