stjames 0.0.32__py3-none-any.whl → 0.0.34__py3-none-any.whl

stjames/_deprecated_solvent_settings.py

@@ -1,4 +1,4 @@
-import pydantic
+from pydantic import PositiveFloat, PositiveInt
 
 from .base import Base, LowercaseStrEnum
 
@@ -14,6 +14,6 @@ class SolventSettings(Base):
     model: ImplicitSolventModel = ImplicitSolventModel.NONE
     epsilon: float = 78.36
 
-    grid_points_per_atom: pydantic.PositiveInt = 170
-    vdw_scale: pydantic.PositiveFloat = 1.2
-    weight_cutoff: pydantic.PositiveFloat = 1e-8
+    grid_points_per_atom: PositiveInt = 170
+    vdw_scale: PositiveFloat = 1.2
+    weight_cutoff: PositiveFloat = 1e-8

stjames/base.py

@@ -1,14 +1,16 @@
 from enum import Enum
-from typing import Annotated, Any, Hashable, TypeVar
+from typing import Annotated, Hashable, TypeVar
 
 import numpy as np
 import pydantic
 
+T = TypeVar("T")
+
 
 class Base(pydantic.BaseModel):
     @pydantic.field_validator("*", mode="before")
     @classmethod
-    def coerce_numpy(cls, val: Any):
+    def coerce_numpy(cls, val: T) -> T | list:
         if isinstance(val, np.ndarray):
             return val.tolist()
         else:
@@ -19,7 +21,8 @@ class LowercaseStrEnum(str, Enum):
     """Enum where hyphens and case are ignored."""
 
     @classmethod
-    def _missing_(cls, value: str):
+    def _missing_(cls, value: str) -> str | None:  # type: ignore
+        # Type note: technically breaking Liskov, value: object in Enum
         for member in cls:
             if member.lower().replace("-", "") == value.lower().replace("-", ""):
                 return member
@@ -27,14 +30,13 @@ class LowercaseStrEnum(str, Enum):
 
 
 # cf. https://github.com/pydantic/pydantic-core/pull/820#issuecomment-1670475909
-# there are python details here I don't really grasp
-T = TypeVar("T", bound=Hashable)
+H = TypeVar("H", bound=Hashable)
 
 
-def _validate_unique_list(v: list[T]) -> list[T]:
+def _validate_unique_list(v: list[H]) -> list[H]:
     if len(v) != len(set(v)):
         raise ValueError("this list must be unique, and isn't!")
     return v
 
 
-UniqueList = Annotated[list[T], pydantic.AfterValidator(_validate_unique_list)]
+UniqueList = Annotated[list[H], pydantic.AfterValidator(_validate_unique_list)]

stjames/basis_set.py

@@ -1,4 +1,5 @@
 import pydantic
+from pydantic import PositiveFloat, PositiveInt
 
 try:
     from typing import Optional, Self
@@ -10,8 +11,8 @@ from .base import Base
 
 class BasisSetOverride(Base):
     name: str
-    atomic_numbers: Optional[list[pydantic.PositiveInt]] = None
-    atoms: Optional[list[pydantic.PositiveInt]] = None  # 1-indexed
+    atomic_numbers: Optional[list[PositiveInt]] = None
+    atoms: Optional[list[PositiveInt]] = None  # 1-indexed
 
     @pydantic.model_validator(mode="after")
     def check_override(self) -> Self:
@@ -29,4 +30,4 @@ class BasisSet(Base):
     # value below which a basis function can be ignored
     # (for improving DFT grid calcs, as per Stratmann/Scuseria/Frisch CPL 1996)
     # this shouldn't really need to be modified...
-    cutoff_threshold: pydantic.PositiveFloat = 1e-10
+    cutoff_threshold: PositiveFloat = 1e-10

stjames/molecule.py

@@ -1,4 +1,5 @@
 import pydantic
+from pydantic import NonNegativeInt, PositiveInt
 
 try:
     from typing import Optional, Self
@@ -16,17 +17,17 @@ class VibrationalMode(Base):
 
 
 class Atom(Base):
-    atomic_number: pydantic.NonNegativeInt
+    atomic_number: NonNegativeInt
     position: list[float]
 
 
 class Molecule(Base):
     charge: int
-    multiplicity: pydantic.PositiveInt
+    multiplicity: PositiveInt
     atoms: list[Atom]
 
     energy: Optional[float] = None
-    scf_iterations: Optional[pydantic.NonNegativeInt] = None
+    scf_iterations: Optional[NonNegativeInt] = None
     scf_completed: Optional[bool] = None
     elapsed: Optional[float] = None
 
@@ -46,40 +47,36 @@ class Molecule(Base):
     thermal_free_energy_corr: Optional[float] = None
 
     @property
-    def coordinates(self):
+    def coordinates(self) -> list[list[float]]:
         return [a.position for a in self.atoms]
 
     @property
-    def atomic_numbers(self):
+    def atomic_numbers(self) -> list[NonNegativeInt]:
         return [a.atomic_number for a in self.atoms]
 
     @property
     def sum_energy_zpe(self) -> Optional[float]:
         if (self.energy is None) or (self.zero_point_energy is None):
             return None
-        else:
-            return self.energy + self.zero_point_energy
+        return self.energy + self.zero_point_energy
 
     @property
     def sum_energy_thermal_corr(self) -> Optional[float]:
         if (self.energy is None) or (self.thermal_energy_corr is None):
             return None
-        else:
-            return self.energy + self.thermal_energy_corr
+        return self.energy + self.thermal_energy_corr
 
     @property
     def sum_energy_enthalpy(self) -> Optional[float]:
         if (self.energy is None) or (self.thermal_enthalpy_corr is None):
             return None
-        else:
-            return self.energy + self.thermal_enthalpy_corr
+        return self.energy + self.thermal_enthalpy_corr
 
     @property
     def sum_energy_free_energy(self) -> Optional[float]:
         if (self.energy is None) or (self.thermal_free_energy_corr is None):
             return None
-        else:
-            return self.energy + self.thermal_free_energy_corr
+        return self.energy + self.thermal_free_energy_corr
 
     @pydantic.model_validator(mode="after")
     def check_electron_sanity(self) -> Self:

stjames/opt_settings.py

@@ -1,4 +1,4 @@
-import pydantic
+from pydantic import Field, PositiveFloat, PositiveInt
 
 from .base import Base, LowercaseStrEnum
 
@@ -19,12 +19,12 @@ class Constraint(Base):
 
 
 class OptimizationSettings(Base):
-    max_steps: pydantic.PositiveInt = 250
+    max_steps: PositiveInt = 250
     transition_state: bool = False
 
     # when are we converged?
-    max_gradient_threshold: pydantic.PositiveFloat = 4.5e-4
-    rms_gradient_threshold: pydantic.PositiveFloat = 3.0e-4
-    energy_threshold: pydantic.PositiveFloat = 1e-6
+    max_gradient_threshold: PositiveFloat = 4.5e-4
+    rms_gradient_threshold: PositiveFloat = 3.0e-4
+    energy_threshold: PositiveFloat = 1e-6
 
-    constraints: list[Constraint] = pydantic.Field(default_factory=list)
+    constraints: list[Constraint] = Field(default_factory=list)

stjames/scf_settings.py

@@ -1,6 +1,7 @@
 from typing import Any, Optional
 
 import pydantic
+from pydantic import PositiveFloat, PositiveInt
 
 from .base import Base, LowercaseStrEnum
 from .diis_settings import DIISSettings
@@ -36,36 +37,36 @@ class SCFSettings(Base):
     #### damping
     do_damping: bool = True
     # when should we stop damping?
-    end_damping_error: pydantic.PositiveFloat = 0.1
+    end_damping_error: PositiveFloat = 0.1
     # what damping factor should we use?
     damping_factor: float = pydantic.Field(ge=0, le=1, default=0.7)
 
     #### level shifting
     do_level_shift: bool = True
     # how much? (Eh)
-    level_shift_magnitude: pydantic.PositiveFloat = 0.25
+    level_shift_magnitude: PositiveFloat = 0.25
     # when should we stop?
-    end_level_shift_error: pydantic.PositiveFloat = 0.1
+    end_level_shift_error: PositiveFloat = 0.1
 
     #### incremental
     # do incremental fock build?
     do_incremental: bool = True
     # reset incremental fock build
-    rebuild_frequency: pydantic.PositiveInt = 20
+    rebuild_frequency: PositiveInt = 20
 
     #### when are we converged?
-    energy_threshold: pydantic.PositiveFloat = 1e-6
-    rms_error_threshold: pydantic.PositiveFloat = 1e-8
-    max_error_threshold: pydantic.PositiveFloat = 1e-5
+    energy_threshold: PositiveFloat = 1e-6
+    rms_error_threshold: PositiveFloat = 1e-8
+    max_error_threshold: PositiveFloat = 1e-5
 
     #### DIIS
     do_diis: bool = True
     # error below which we'll start DIIS
     start_diis_max_error: pydantic.PositiveFloat = 0.2
     # first iteration we'll consider starting DIIS
-    start_diis_iter: pydantic.PositiveInt = 3
+    start_diis_iter: PositiveInt = 3
     # iteration past which we'll start DIIS even if error is high
-    start_diis_anyway: pydantic.PositiveInt = 7
+    start_diis_anyway: PositiveInt = 7
 
     # if ``read`` initialization is selected
     initial_density_matrix_guess: Optional[list[list[float]]] = None

stjames/settings.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional
+from typing import Any, Optional, TypeVar
 
 import pydantic
 
@@ -13,7 +13,14 @@ from .solvent import SolventSettings
 from .task import Task
 from .thermochem_settings import ThermochemistrySettings
 
-PREPACKAGED_METHODS = [Method.HF3C, Method.B973C, Method.AIMNET2_WB97MD3, Method.AIMNET2_B973C, Method.GFN2_XTB, Method.GFN1_XTB]
+PREPACKAGED_METHODS = [
+    Method.HF3C,
+    Method.B973C,
+    Method.AIMNET2_WB97MD3,
+    Method.AIMNET2_B973C,
+    Method.GFN2_XTB,
+    Method.GFN1_XTB,
+]
 
 METHODS_WITH_CORRECTION = [
     Method.B97D3,
@@ -23,6 +30,8 @@ METHODS_WITH_CORRECTION = [
     Method.WB97MV,
 ]
 
+T = TypeVar("T")
+
 
 class Settings(Base):
     method: Method = Method.HARTREE_FOCK
@@ -39,8 +48,8 @@ class Settings(Base):
     opt_settings: OptimizationSettings = OptimizationSettings()
     thermochem_settings: ThermochemistrySettings = ThermochemistrySettings()
 
-    @pydantic.computed_field
     @property
+    @pydantic.computed_field
     def level_of_theory(self) -> str:
         if self.method in PREPACKAGED_METHODS or self.basis_set is None:
             method = self.method.value
@@ -72,18 +81,17 @@ class Settings(Base):
 
     @pydantic.field_validator("basis_set", mode="before")
     @classmethod
-    def parse_basis_set(cls, v: Any) -> Optional[BasisSet]:
+    def parse_basis_set(cls, v: Any) -> BasisSet | dict | None:
         """Turn a string into a ``BasisSet`` object. (This is a little crude.)"""
         if isinstance(v, BasisSet):
             return None if v.name is None else v
         elif isinstance(v, dict):
-            return None if v["name"] is None else v
+            return None if v.get("name") is None else v
         elif isinstance(v, str):
             if len(v):
                 return BasisSet(name=v)
-            else:
-                # "" is basically None, let's be real here...
-                return None
+            # "" is basically None, let's be real here...
+            return None
         elif v is None:
             return None
         else:
@@ -91,7 +99,7 @@ class Settings(Base):
 
     @pydantic.field_validator("corrections", mode="before")
     @classmethod
-    def remove_empty_string(cls, v: list) -> list:
+    def remove_empty_string(cls, v: list[T]) -> list[T]:
         """Remove empty string values."""
         return [c for c in v if c]
 

stjames/workflows/pka.py

@@ -1,15 +1,15 @@
 from typing import Optional
 
-from ..mode import Mode
 from ..base import Base
+from ..mode import Mode
 from .workflow import DBCalculation, Workflow
 
 
 class pKaMicrostate(Base):
     atom_index: int
     structures: list[DBCalculation] = []
-    delta_G: float
-    pKa: float
+    deltaG: float
+    pka: float
 
 
 class pKaWorkflow(Workflow):

stjames/workflows/redox_potential.py

@@ -1,4 +1,4 @@
-from typing import Optional, Any
+from typing import Any, Optional
 
 from ..mode import Mode
 from ..solvent import Solvent
@@ -11,7 +11,9 @@ class RedoxPotentialWorkflow(Workflow):
     reduction: bool = True
     oxidation: bool = True
 
+    # legacy values - remove in future release!
     redox_type: Optional[str] = None
+    redox_potential: Optional[float] = None
 
     # uuids
     neutral_molecule: Optional[str] = None
@@ -22,10 +24,12 @@ class RedoxPotentialWorkflow(Workflow):
     oxidation_potential: Optional[float] = None
 
     def model_post_init(self, __context: Any) -> None:
-        """ Keep back-compatible with old schema. """
+        """Keep back-compatible with old schema."""
         if self.redox_type == "oxidation":
             self.oxidation = True
             self.reduction = False
+            self.oxidation_potential = self.redox_potential
         elif self.redox_type == "reduction":
             self.oxidation = False
             self.reduction = True
+            self.reduction_potential = self.redox_potential

stjames/workflows/tautomer.py

@@ -2,7 +2,6 @@ from typing import Optional
 
 from ..base import Base
 from ..mode import Mode
-from .conformer import ConformerSettings
 from .workflow import DBCalculation, Workflow
 
 

{stjames-0.0.32.dist-info → stjames-0.0.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stjames
-Version: 0.0.32
+Version: 0.0.34
 Summary: standardized JSON atom/molecule encoding scheme
 Author-email: Corin Wagen <corin@rowansci.com>
 Project-URL: Homepage, https://github.com/rowansci/stjames
@@ -9,6 +9,7 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pydantic >=2.4
+Requires-Dist: numpy
 
 # stjames
 

{stjames-0.0.32.dist-info → stjames-0.0.34.dist-info}/RECORD

@@ -1,7 +1,7 @@
 stjames/__init__.py,sha256=_gX08PEdPHoE6oyc0OF2gKPBfk9op0am75y1TDUo-MU,451
-stjames/_deprecated_solvent_settings.py,sha256=ALUZeQoxDS11EvZexT1J2SDAlLTiXe-x4jV8tenDqMc,445
-stjames/base.py,sha256=cZNJq0TSRJk7aKVIl7fzRG6IYQeJ0ihWPi9eJxTiQfI,1070
-stjames/basis_set.py,sha256=4Su3K3Y7G-4S8WHWOQE94d6hxoN6L_MkQUz3qoFqcB0,1014
+stjames/_deprecated_solvent_settings.py,sha256=gj5j9p3zakIwSTK5_ndqBXJx--IzjZNxZ75z-wipLOo,450
+stjames/base.py,sha256=GTISjYPBi4KCxAR9DPLf4mp6ZYuh2MwlNI0jZ1N41Io,1142
+stjames/basis_set.py,sha256=Fi6vJx-ni2aOozFYH2hrSlc9lL01XzPO4zO6onJliuE,1035
 stjames/calculation.py,sha256=6xnoKLn6tr1QaactTWsdu8Ciqdu8oa-hmB2y9RIrJTI,786
 stjames/correction.py,sha256=_pNG3qSylfx0iyUxqwx9HPU0m032YwP6wSPCjbJrD94,358
 stjames/diis_settings.py,sha256=QHc7L-hktkbOWBYr29byTdqL8lWJzKJiY9XW8ha4Qzo,552
@@ -9,10 +9,11 @@ stjames/grid_settings.py,sha256=WrSNGc-8_f87YBZYt9Hh7RbhM4MweADoVzwBMcSqcsE,640
 stjames/int_settings.py,sha256=5HXp8opt5ZyY1UpmfaK7NVloWVLM5jkG0elEEqpVLUo,896
 stjames/method.py,sha256=e7A3XH808JglgfCzISKyRKAxGDj7BQdcQQtgtTlgfNk,843
 stjames/mode.py,sha256=xw46Cc7f3eTS8i35qECi-8DocAlANhayK3w4akD4HBU,496
-stjames/molecule.py,sha256=y-NidMubuA9IsSruUKavyBndbf5hYodjihna9uBjkOc,2915
-stjames/opt_settings.py,sha256=RwXikvwW6fTODK3FK8Xvyqy4DY-qMVtZHndWirBxujk,735
-stjames/scf_settings.py,sha256=blJ5Mo2znF_H05gBCPZXTrDwCZuVzUJ_OIcmYGc0Eg0,2389
-stjames/settings.py,sha256=xw5m9Zq4L-oUoV53MMmn9t-KmzlZoFVJSRmac8f2Sfw,8420
+stjames/molecule.py,sha256=t0jOmkTMgQE2ayBn9igjpY_4gPyqLPqZLlgtwd4uBOA,2910
+stjames/opt_settings.py,sha256=81vYDBEbqLGN_crI1GtAxiOoRa791WN0m2HfT5_A7pE,729
+stjames/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+stjames/scf_settings.py,sha256=xMMCQ0hVB4nNFSiWesNQZUa_aLsozSZGYWweAPPDGBg,2356
+stjames/settings.py,sha256=cUiAydIVes9iDEky4coOYQ2kF97AYaLClx5qjJmEpWY,8462
 stjames/solvent.py,sha256=3ssY-vQCCa071srfJbwCYEKDpWos8Mgvmg_dTpAnbJw,1104
 stjames/status.py,sha256=wTKNcNxStoEHrxxgr_zTyN90NITa3rxMQZzOgrCifEw,332
 stjames/task.py,sha256=TTl-iTdvDNCZTdPsyS6bYmxzY0ez9PgYlL62fztayXQ,307
@@ -20,13 +21,13 @@ stjames/thermochem_settings.py,sha256=ZTLz31v8Ltutde5Nfm0vH5YahWjcfFWfr_R856Kffx
 stjames/workflows/__init__.py,sha256=TibQzeD3X3YGq9TaAzPlBMxU7zskFoa8PwPUam4TcZY,154
 stjames/workflows/conformer.py,sha256=tqACquXy8wqIbxl3T-n4dfNNNkVdLos2dIG1u0CV8cc,2169
 stjames/workflows/fukui.py,sha256=FAabgLi_ig6Zp_jiTkuveP0wAObwv68liuumr85rb0g,366
-stjames/workflows/pka.py,sha256=TdTZqCBOL6DnPnuGs7qlCVt21LVSveryP0h1hz_wBRU,775
-stjames/workflows/redox_potential.py,sha256=uQZvgzii9SiYYxwzUTlb43uCSVaQ5C5Sg52HJ3YPhcI,897
+stjames/workflows/pka.py,sha256=zpR90Yv2L-D56o2mGArM8027DWpnFFnay31UR9Xh5Nc,774
+stjames/workflows/redox_potential.py,sha256=5CApMfltVRrPKilGwW1_rKhA5HoL75ZSIOCVv90aN6k,1107
 stjames/workflows/scan.py,sha256=d1ca1uk1nxdzlSwqMLZ0W1_DHyrQMnHJmvPfziCS0SA,773
-stjames/workflows/tautomer.py,sha256=xJMDArpsuTLmgBZHXcBGVMcP_tncQzvxUH4Kbis4opg,462
+stjames/workflows/tautomer.py,sha256=kZSCHo2Q7LzqtQjF_WyyxjECkndG49T9QOM12hsUkx8,421
 stjames/workflows/workflow.py,sha256=jzkHO8mdmoy9zW3iD457KAXYrCCkV2OHjg6T3896nCQ,304
-stjames-0.0.32.dist-info/LICENSE,sha256=i7ehYBS-6gGmbTcgU4mgk28pyOx2kScJ0kcx8n7bWLM,1084
-stjames-0.0.32.dist-info/METADATA,sha256=mtFI6nQUc_03w1bFfF7CKnyCoSCmhDFJKNTko8dLvw0,1605
-stjames-0.0.32.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-stjames-0.0.32.dist-info/top_level.txt,sha256=FYCwxl6quhYOAgG-mnPQcCK8vsVM7B8rIUrO-WrQ_PI,8
-stjames-0.0.32.dist-info/RECORD,,
+stjames-0.0.34.dist-info/LICENSE,sha256=i7ehYBS-6gGmbTcgU4mgk28pyOx2kScJ0kcx8n7bWLM,1084
+stjames-0.0.34.dist-info/METADATA,sha256=Yv7Q-jpvJKU1wcU5VELqXUNcJks_EpOeDZBFDytihZc,1626
+stjames-0.0.34.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+stjames-0.0.34.dist-info/top_level.txt,sha256=FYCwxl6quhYOAgG-mnPQcCK8vsVM7B8rIUrO-WrQ_PI,8
+stjames-0.0.34.dist-info/RECORD,,