stjames 0.0.102__py3-none-any.whl → 0.0.103__py3-none-any.whl

stjames/engine.py

@@ -4,12 +4,11 @@ from .base import LowercaseStrEnum
 class Engine(LowercaseStrEnum):
     AIMNET2 = "aimnet2"
     MACE = "mace"
-    OCP24 = "ocp24"
     OMOL25 = "omol25"
     ORB = "orb"
     TBLITE = "tblite"
     XTB = "xtb"
-    TERRACHEM = "terrachem"
+    TERACHEM = "terachem"
     PYSCF = "pyscf"
     PSI4 = "psi4"
     OPENFF = "openff"

stjames/method.py

@@ -34,12 +34,11 @@ class Method(LowercaseStrEnum):
     AIMNET2_WB97MD3 = "aimnet2_wb97md3"
     MACE_MP_0 = "mace_mp_0"
     MACE_MP_0B2_L = "mace_mp_0b2_l"
-    OCP24_S = "ocp24_s"
-    OCP24_L = "ocp24_l"
     OMOL25_CONSERVING_S = "omol25_conserving_s"
     UMA_M_OMOL = "uma_m_omol"
     UMA_S_OMOL = "uma_s_omol"
     ORB_V3_CONSERVATIVE_INF_OMAT = "orb_v3_conservative_inf_omat"
+    ORB_V3_CONSERVATIVE_OMOL = "orb_v3_conservative_omol"
 
     GFN_FF = "gfn_ff"
     GFN0_XTB = "gfn0_xtb"
@@ -77,11 +76,9 @@ class Method(LowercaseStrEnum):
                 return Engine.AIMNET2
             case Method.MACE_MP_0B2_L:
                 return Engine.MACE
-            case Method.OCP24_S | Method.OCP24_L:
-                return Engine.OCP24
             case Method.OMOL25_CONSERVING_S | Method.UMA_M_OMOL | Method.UMA_S_OMOL:
                 return Engine.OMOL25
-            case Method.ORB_V3_CONSERVATIVE_INF_OMAT:
+            case Method.ORB_V3_CONSERVATIVE_INF_OMAT | Method.ORB_V3_CONSERVATIVE_OMOL:
                 return Engine.ORB
             case method if method in XTB_METHODS:
                 return Engine.TBLITE if is_periodic else Engine.XTB
@@ -96,12 +93,11 @@ class Method(LowercaseStrEnum):
 
 PrepackagedNNPMethod = Literal[
     Method.AIMNET2_WB97MD3,
-    Method.OCP24_S,
-    Method.OCP24_L,
     Method.OMOL25_CONSERVING_S,
     Method.UMA_M_OMOL,
     Method.UMA_S_OMOL,
     Method.ORB_V3_CONSERVATIVE_INF_OMAT,
+    Method.ORB_V3_CONSERVATIVE_OMOL,
     Method.EGRET_1,
     Method.EGRET_1E,
     Method.EGRET_1T,
@@ -109,12 +105,11 @@ PrepackagedNNPMethod = Literal[
 
 PREPACKAGED_NNP_METHODS = [
     Method.AIMNET2_WB97MD3,
-    Method.OCP24_S,
-    Method.OCP24_L,
     Method.OMOL25_CONSERVING_S,
     Method.UMA_M_OMOL,
     Method.UMA_S_OMOL,
     Method.ORB_V3_CONSERVATIVE_INF_OMAT,
+    Method.ORB_V3_CONSERVATIVE_OMOL,
     Method.EGRET_1,
     Method.EGRET_1E,
     Method.EGRET_1T,

stjames/types.py

@@ -1,4 +1,4 @@
-from typing import Callable, Iterable, TypeAlias
+from typing import Callable, Iterable, Protocol, TypeAlias, overload
 
 UUID: TypeAlias = str
 
@@ -10,22 +10,32 @@ FloatPerAtom: TypeAlias = list[float]
 Matrix3x3: TypeAlias = tuple[Vector3D, Vector3D, Vector3D]
 
 
-def round_list(round_to: int = 6) -> Callable[[Iterable[float]], list[float]]:
-    """Create a validator that rounds each float in a list to a given number of decimal places."""
+class _Iterable_Rounder(Protocol):
+    """Callable with return types conditioned on the input type(s)."""
 
-    def rounder(values: Iterable[float]) -> list[float]:
-        return [round(value, round_to) for value in values]
+    @overload
+    def __call__(self, values: None, /) -> None: ...
+    @overload
+    def __call__(self, values: Iterable[float], /) -> list[float]: ...
+    @overload
+    def __call__(self, values: Iterable[float | None], /) -> list[float | None]: ...
 
-    return rounder
 
+def round_list(round_to: int = 6) -> _Iterable_Rounder:
+    """Round values (even if Nones), preserving types."""
 
-def round_optional_list(round_to: int = 6) -> Callable[[Iterable[float] | None], list[float] | None]:
-    """Create a validator that rounds each float in a list to a given number of decimal places."""
+    @overload
+    def rounder(values: None, /) -> None: ...
+    @overload
+    def rounder(values: Iterable[float], /) -> list[float]: ...
+    @overload
+    def rounder(values: Iterable[float | None], /) -> list[float | None]: ...
 
-    def rounder(values: Iterable[float] | None) -> list[float] | None:
+    def rounder(values: Iterable[float] | Iterable[float | None] | None, /) -> list[float] | list[float | None] | None:
         if values is None:
             return None
-        return [round(value, round_to) for value in values]
+
+        return [round(v, round_to) if v is not None else None for v in values]
 
     return rounder
 

stjames/workflows/conformer_search.py

@@ -1,7 +1,7 @@
 """Conformer search workflow."""
 
 from abc import ABC
-from typing import Annotated, Self, Sequence, TypeVar
+from typing import Annotated, Literal, Self, Sequence, TypeVar
 
 from pydantic import AfterValidator, BaseModel, Field, field_validator, model_validator
 
@@ -94,6 +94,7 @@ class ETKDGSettings(ConformerGenSettings):
     num_confs_considered: int = 100
     max_mmff_iterations: int = 500
     max_mmff_energy: float | None = 30
+    settings_type: Literal["etkdg"] = "etkdg"
 
     @field_validator("constraints")
     def check_constraints(cls, constraints: Sequence[Constraint]) -> Sequence[Constraint]:
@@ -209,6 +210,7 @@ class iMTDSettings(ConformerGenSettings, ABC):
     speed: iMTDSpeeds = iMTDSpeeds.QUICK
     reopt: bool = _sentinel  # type: ignore [assignment]
     free_energy_weights: bool = False
+    settings_type: Literal["imtd"] = "imtd"
 
     @model_validator(mode="after")
     def validate_and_build_imtdgc_settings(self) -> Self:
@@ -253,6 +255,9 @@ class iMTDsMTDSettings(iMTDSettings):
     run_type: str = "imtd-smtd"
 
 
+ConformerGenSettingsUnion = Annotated[ETKDGSettings | iMTDSettings, Field(discriminator="settings_type")]
+
+
 class ConformerGenMixin(BaseModel):
     """
     Mixin for workflows that need conformer generation.
@@ -265,7 +270,7 @@ class ConformerGenMixin(BaseModel):
     """
 
     conf_gen_mode: Mode = Mode.RAPID
-    conf_gen_settings: ConformerGenSettings = _sentinel  # type: ignore [assignment]
+    conf_gen_settings: ConformerGenSettingsUnion = _sentinel  # type: ignore [assignment]
     constraints: Sequence[Constraint] = tuple()
     nci: bool = False
     max_confs: int | None = None
@@ -273,8 +278,8 @@ class ConformerGenMixin(BaseModel):
     @model_validator(mode="after")
     def validate_and_build_conf_gen_settings(self) -> Self:
         """Validate and build the ConformerGenSettings."""
-        if self.conf_gen_settings is not _sentinel and self.conf_gen_mode != Mode.MANUAL:
-            raise ValueError("Cannot specify conf_gen_settings with non-MANUAL mode")
+        if self.conf_gen_settings is not _sentinel:
+            return self
 
         match self.conf_gen_mode:
             case Mode.MANUAL:

stjames/workflows/docking.py

@@ -7,7 +7,7 @@ from pydantic import AfterValidator, ConfigDict, field_validator, model_validato
 from ..base import Base, round_float
 from ..pdb import PDB
 from ..types import UUID, Vector3D
-from .conformer_search import ConformerGenSettings, ETKDGSettings
+from .conformer_search import ConformerGenSettingsUnion, ETKDGSettings
 from .workflow import MoleculeWorkflow
 
 
@@ -79,12 +79,12 @@ class DockingWorkflow(MoleculeWorkflow):
     pocket: tuple[Vector3D, Vector3D]
 
     do_csearch: bool = True
-    conformer_gen_settings: ConformerGenSettings = ETKDGSettings(mode="reckless")
+    conformer_gen_settings: ConformerGenSettingsUnion = ETKDGSettings(mode="reckless")
 
     do_optimization: bool = True
     # optimization_settings - here in future once we have a cleaner mode sol'n, ccw 7.9.25
 
-    docking_settings: DockingSettings = VinaSettings()
+    docking_settings: VinaSettings = VinaSettings()
 
     do_pose_refinement: bool = True
 

stjames/workflows/macropka.py

@@ -4,7 +4,7 @@ from typing import Annotated, Optional, Self
 
 from pydantic import AfterValidator, model_validator
 
-from ..base import Base, round_float
+from ..base import Base, round_float, round_optional_float
 from ..types import round_list
 from .workflow import SMILESWorkflow
 
@@ -72,9 +72,9 @@ class MacropKaWorkflow(SMILESWorkflow):
 
     microstates: list[MacropKaMicrostate] = []
     pKa_values: list[MacropKaValue] = []
-    isoelectric_point: Annotated[Optional[float], AfterValidator(round_float(3))] = None
-    solvation_energy: Annotated[Optional[float], AfterValidator(round_float(3))] = None
-    kpuu_probability: Annotated[Optional[float], AfterValidator(round_float(3))] = None
+    isoelectric_point: Annotated[Optional[float], AfterValidator(round_optional_float(3))] = None
+    solvation_energy: Annotated[Optional[float], AfterValidator(round_optional_float(3))] = None
+    kpuu_probability: Annotated[Optional[float], AfterValidator(round_optional_float(3))] = None
 
     microstate_weights_by_pH: list[
         tuple[

stjames/workflows/multistage_opt.py

@@ -1,5 +1,6 @@
 """Multi-stage optimization workflow."""
 
+import logging
 import re
 from typing import Self, Sequence
 
@@ -18,6 +19,8 @@ from ..task import Task
 from ..types import UUID
 from .workflow import MoleculeWorkflow
 
+logger = logging.getLogger(__name__)
+
 
 class MultiStageOptSettings(BaseModel):
     """
@@ -109,7 +112,8 @@ class MultiStageOptSettings(BaseModel):
                 pass
 
             case (mode, True):
-                raise ValueError(f"Cannot specify optimization_settings or singlepoint_settings with {mode=}")
+                logger.debug(f"Mode {mode=} specified with optimization_settings or singlepoint_settings, ignoring")
+                pass
 
             case (mode, False):
                 self._assign_settings_by_mode(mode)
@@ -253,7 +257,8 @@ class MultiStageOptMixin(BaseModel):
                 pass
 
             case (mso_mode, msos) if msos is not _sentinel_msos:
-                raise ValueError(f"Cannot specify multistage_opt_settings with {mso_mode=}, {msos=}")
+                logger.debug(f"Mode {mso_mode=} specified with multistage_opt_settings, ignoring")
+                pass
 
             case (mso_mode, _):
                 self.multistage_opt_settings = MultiStageOptSettings(

stjames/workflows/nmr.py

@@ -8,7 +8,7 @@ from ..base import Base, LowercaseStrEnum, round_float
 from ..mode import Mode
 from ..settings import Settings
 from ..solvent import Solvent
-from ..types import UUID, round_list, round_optional_list
+from ..types import UUID, round_list
 from .conformer_search import ConformerGenSettings, iMTDSettings
 from .multistage_opt import MultiStageOptSettings
 from .workflow import MoleculeWorkflow
@@ -66,8 +66,8 @@ class NMRSpectroscopyWorkflow(MoleculeWorkflow):
 
     conformers: list[UUID] = []
     boltzmann_weights: Annotated[list[float], AfterValidator(round_list(3))] = []
-    per_conformer_chemical_shifts: list[Annotated[list[float | None], AfterValidator(round_optional_list(3))]] = []
-    chemical_shifts: Annotated[list[float | None], AfterValidator(round_optional_list(3))] = []
+    per_conformer_chemical_shifts: list[Annotated[list[float | None], AfterValidator(round_list(3))]] = []
+    chemical_shifts: Annotated[list[float | None], AfterValidator(round_list(3))] = []
     symmetry_equivalent_nuclei: list[list[int]] = []
 
     predicted_peaks: dict[int, list[NMRPeak]] = {}

stjames/workflows/protein_cofolding.py

@@ -5,7 +5,7 @@ from typing import Annotated, Literal
 from pydantic import AfterValidator, BaseModel, ConfigDict
 
 from ..base import LowercaseStrEnum, round_float
-from ..types import UUID, round_optional_list
+from ..types import UUID, round_list
 from .workflow import FASTAWorkflow
 
 
@@ -86,4 +86,4 @@ class ProteinCofoldingWorkflow(FASTAWorkflow):
     scores: CofoldingScores | None = None
     model: CofoldingModel = CofoldingModel.BOLTZ_2
     affinity_score: AffinityScore | None = None
-    lddt: Annotated[list[float] | None, AfterValidator(round_optional_list(3))] = None
+    lddt: Annotated[list[float] | None, AfterValidator(round_list(3))] = None

stjames/workflows/solubility.py

@@ -5,7 +5,7 @@ from typing import Annotated, Self
 from pydantic import AfterValidator, BaseModel, model_validator
 
 from ..base import LowercaseStrEnum
-from ..types import round_list, round_optional_list
+from ..types import round_list
 from .workflow import SMILESWorkflow
 
 
@@ -24,7 +24,7 @@ class SolubilityResult(BaseModel):
     """
 
     solubilities: Annotated[list[float], AfterValidator(round_list(6))]
-    uncertainties: Annotated[list[float | None], AfterValidator(round_optional_list(6))]
+    uncertainties: Annotated[list[float | None], AfterValidator(round_list(6))]
 
     @model_validator(mode="after")
     def check_size(self) -> Self:

{stjames-0.0.102.dist-info → stjames-0.0.103.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stjames
-Version: 0.0.102
+Version: 0.0.103
 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.102.dist-info → stjames-0.0.103.dist-info}/RECORD

@@ -7,9 +7,9 @@ stjames/calculation.py,sha256=O2LwwQ_cOLmDOGXTHA9J71YbUZXigUSbvbLA-fSVm3w,915
 stjames/compute_settings.py,sha256=wuYE6W4WqP3oFHwAeEp7XQ0oMz_MNLucWgrsuJ4kEFQ,268
 stjames/constraint.py,sha256=B6oV0rYjmAWr8gpi5f03gRy_uuqjUURVDVwoez5Cfbg,2442
 stjames/correction.py,sha256=ZVErCcj4TPyZeKrdvXVjHa0tFynsCaoy96QZUVxWFM8,413
-stjames/engine.py,sha256=4Vt1uF1jNGPj-06F2I73oU5RoH1MlNOlDRSNkWN91NE,315
+stjames/engine.py,sha256=jaOHi3nhKZG85HAvKu-HuDl1Hh9nnJfhmOez2ghZXWA,293
 stjames/message.py,sha256=Rq6QqmHZKecWxYH8fVyXmuoCCPZv8YinvgykSeorXSU,216
-stjames/method.py,sha256=gZbrHAivm2OXCbF3_6swei1LG6YQ7Uo1EY7aN0x3XHs,4501
+stjames/method.py,sha256=tMdngVD7kU5ob285YD2-2UVq1gwf0T7dkGF0qYSNoWQ,4453
 stjames/mode.py,sha256=xw46Cc7f3eTS8i35qECi-8DocAlANhayK3w4akD4HBU,496
 stjames/molecule.py,sha256=4Tsa_gRC4C-qGPop24jRIs2q07r0LGbSRb-ApOaZa5c,20839
 stjames/opt_settings.py,sha256=LEwGXUEKq5TfU5rr60Z4QQBhCqiw1Ch5w0M_lXawWo8,642
@@ -22,7 +22,7 @@ stjames/solvent.py,sha256=u037tmu-9oa21s-WEDZ7VC7nuNVjkqR2ML4JWjWSME4,1158
 stjames/status.py,sha256=KQHDqWSd4kBLow23YLcfOkFdtqN61RFZI-jf2zANWRY,501
 stjames/task.py,sha256=OLINRqe66o7t8arffilwmggrF_7TH0L79u6DhGruxV8,329
 stjames/thermochem_settings.py,sha256=ZTLz31v8Ltutde5Nfm0vH5YahWjcfFWfr_R856KffxE,517
-stjames/types.py,sha256=Fco-Lw8DYvUhbSvOjYZHaE_AaMyFNdWLUjNvb5aTAzw,4498
+stjames/types.py,sha256=PGEZeqGnomgWUSHcMuUftzvZt7l4VwaQ1Pi-7vEkibg,4773
 stjames/atomium_stjames/__init__.py,sha256=gZkzC7i9D_fmWUTN55gtygITo3-qvJUda5CXLR0jyCQ,306
 stjames/atomium_stjames/data.py,sha256=-hzwBpTHq5JetsOVyopUJswKnKAkMtJ_XkONxjXVupU,5675
 stjames/atomium_stjames/mmcif.py,sha256=em1fNt6577OaUjL7Pctru7aJp3ceZ9kEnj5w6BRWdVs,27090
@@ -42,30 +42,30 @@ stjames/workflows/admet.py,sha256=h8ph6oeRCxU3-_jqRRWPg3RZcheu9JzCHiWqSC9VYKY,12
 stjames/workflows/basic_calculation.py,sha256=sAgHBcNHE72ZbZPB9vyZShALRC4zOVw6It6cpJlbX2A,911
 stjames/workflows/bde.py,sha256=g_In-caftXiimrhfdptHjpfrYQUs3vF58qYmRnaTN8g,10825
 stjames/workflows/conformer.py,sha256=18aO6ngMBeGAmQkBdLGCCHr398RIYr1v2hD2IT1u4cc,3005
-stjames/workflows/conformer_search.py,sha256=4yPEKIIedeaVvaAwgjwC1FxiHqM6n2zOF6c9Yk_q1oA,13513
+stjames/workflows/conformer_search.py,sha256=clWFstrz-y1N1x8KtbY7kYUdLrWFSlTLKr-9KnJxt8M,13625
 stjames/workflows/descriptors.py,sha256=T4tc7xdtBdxESGO86KR323jPQ2pgwxBqgV0khA6MEgQ,584
-stjames/workflows/docking.py,sha256=Lc3xB2H_hu0bxlGQBurDtytZfVwSFCgcTyJFWjMDhKI,3857
+stjames/workflows/docking.py,sha256=JG-c6um2tVNeu5H1uzpdQEM5RGMXKgcGNKptTstqj9o,3864
 stjames/workflows/double_ended_ts_search.py,sha256=ovJgEVFc6c3mijCE3TKAY70YvqNmAZ5Y4XgV4-tIxBI,3127
 stjames/workflows/electronic_properties.py,sha256=GT3-NC7w-dbcOJ-3AzJ7LgzH6frTbiH2Iyb9BCa-SvY,4112
 stjames/workflows/fukui.py,sha256=095GDGSSEc5PDD1aoKM8J7icgR5tfwS5Bs9XxFQHge4,2387
 stjames/workflows/hydrogen_bond_basicity.py,sha256=q9eXty68ZyCmrB6G_8bfeOT8Ui_IQquRPu6z-3rNreQ,1589
 stjames/workflows/ion_mobility.py,sha256=rvvAt4ZD-Npl0tHL7TEuaS3XKjHiMkSZVzyJoivdyps,1665
 stjames/workflows/irc.py,sha256=ZP7icylW8rgo_Uh7h3bmyumn0ru1IyF-61nP5Jnmq3M,3402
-stjames/workflows/macropka.py,sha256=KRIyk4gsSYL3eqyzCDndStGLwjWSo60cgCAzvAoD1Nk,3754
+stjames/workflows/macropka.py,sha256=Krj0xXuB-u57Kqlf4bbRiHDUWCpliFr6YPiYqPmYaWk,3803
 stjames/workflows/molecular_dynamics.py,sha256=HqWNxxPSAphfI0DdbTERFkq8UeBjEvhnA_ETv0xw_RY,3522
-stjames/workflows/multistage_opt.py,sha256=SpbA8hNvktnlLS7C-9mBNGluBSrdVS8ygHl1C4TzWcI,16499
-stjames/workflows/nmr.py,sha256=Fw7k_ZvIaxpAxkdgW2xMORhaI5s5BY_v7_VEPG1Ylgc,2710
+stjames/workflows/multistage_opt.py,sha256=UN-4WLsT2WEjO5KqDPrcCkb708Co-ZScHx3g2bto768,16597
+stjames/workflows/nmr.py,sha256=2314qzkAFt-Wu6IDC2AI7eedYocqcCxIy9uZWkHMz_Q,2671
 stjames/workflows/pka.py,sha256=j3vBh2YM3nJzJ1XJKPsmYahRCeaU9n3P-G-u9_moaFw,2065
 stjames/workflows/pose_analysis_md.py,sha256=ES0XlzaLpTjhLrNvcB0zFZa1b1ZHXekN72EbLsx0Skw,4723
-stjames/workflows/protein_cofolding.py,sha256=6WWbVOGzFjIRgQLMmMVODfIxo1jYFmRS1xnJmN0kOBw,3016
+stjames/workflows/protein_cofolding.py,sha256=ZjHER5DsO2vgCpgqttcNIU8xISN5B4DdqdrTxLKFsKY,2998
 stjames/workflows/redox_potential.py,sha256=7S18t9Y3eynSnA3lZbRlvLfdbgeBopdiigLzt1zxg5c,3871
 stjames/workflows/scan.py,sha256=DXQBpa2t2PowAtOwmdgpxaSLq--fEShljzAGSb8Nf5U,2993
-stjames/workflows/solubility.py,sha256=C5FD16p3bkvJz9rhZkgrg_gMrYGY4mnqvXPqqLbixiw,2991
+stjames/workflows/solubility.py,sha256=DhsOxThAslWihpvOKJINNeMiBWSEzRsdqBlfgo2PwX4,2961
 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.102.dist-info/licenses/LICENSE,sha256=i05z7xEhyrg6f8j0lR3XYjShnF-MJGFQ-DnpsZ8yiVI,1084
-stjames-0.0.102.dist-info/METADATA,sha256=sRGYL13y6ucmCcnYzhSzmmClIOn-uyzvURPkABaJ8Lo,1725
-stjames-0.0.102.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-stjames-0.0.102.dist-info/top_level.txt,sha256=FYCwxl6quhYOAgG-mnPQcCK8vsVM7B8rIUrO-WrQ_PI,8
-stjames-0.0.102.dist-info/RECORD,,
+stjames-0.0.103.dist-info/licenses/LICENSE,sha256=i05z7xEhyrg6f8j0lR3XYjShnF-MJGFQ-DnpsZ8yiVI,1084
+stjames-0.0.103.dist-info/METADATA,sha256=ncWtuaR7uYzpmbJ89kySb8O5pJwKJDFX9jBQ5OXl9GY,1725
+stjames-0.0.103.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+stjames-0.0.103.dist-info/top_level.txt,sha256=FYCwxl6quhYOAgG-mnPQcCK8vsVM7B8rIUrO-WrQ_PI,8
+stjames-0.0.103.dist-info/RECORD,,