pydantic v2 api

.github/workflows/CI.yml

@@ -101,10 +101,11 @@ jobs:
         qcore --accept-license
 
     # note: psi4 on c-f pins to a single qcel and qcng, so this may be handy for solve-and-replace
-    #- name: Special Config - QCElemental Dep
+    - name: Special Config - QCElemental Dep
     #  if: (matrix.cfg.label == 'ADCC')
-    #  run: |
-    #    conda remove qcelemental --force
+      run: |
+        conda remove qcelemental --force
+        python -m pip install git+https://github.com/Lnaden/QCElemental.git@Pydanticv2Overhaul --no-deps
     #    python -m pip install qcelemental>=0.26.0 --no-deps
 
       # note: conda remove --force, not mamba remove --force b/c https://github.com/mamba-org/mamba/issues/412

devtools/conda-envs/adcc.yaml

@@ -13,7 +13,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.24.0
-  - pydantic=1
+  - pydantic>=2.1
+  - pydantic-settings
   - msgpack-python
 
     # Testing

devtools/conda-envs/base.yaml

@@ -8,7 +8,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.12.0
-  - pydantic>=1.0.0
+  - pydantic>=2.1
+  - pydantic-settings
 
     # Testing
   - pytest

devtools/conda-envs/docs-cf.yaml

@@ -5,7 +5,8 @@ channels:
 dependencies:
   - python
   - networkx
-  - pydantic=1
+  - pydantic>=2.1
+  - pydantic-settings
   - numpy
   - pint
 

devtools/conda-envs/mrchem.yaml

@@ -13,7 +13,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental>=0.24
-  - pydantic 
+  - pydantic>=2.1
+  - pydantic-settings
 
     # Testing
   - pytest

devtools/conda-envs/nwchem.yaml

@@ -8,7 +8,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.24.0
-  - pydantic>=1.0.0
+  - pydantic>=2.1
+  - pydantic-settings
   - networkx>=2.4.0
 
     # Testing

devtools/conda-envs/openmm.yaml

@@ -16,7 +16,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.11.1
-  - pydantic >=1.8.2
+  - pydantic>=2.1
+  - pydantic-settings
   - pint <0.22
 
     # Testing

devtools/conda-envs/opt-disp.yaml

@@ -25,7 +25,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.26.0
-  - pydantic>=1.0.0
+  - pydantic>=2.1
+  - pydantic-settings
   - msgpack-python
 
     # Testing

devtools/conda-envs/psi-nightly.yaml

@@ -12,7 +12,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.26.0
-  - pydantic>=1.0.0
+  - pydantic>=2.1
+  - pydantic-settings
   - msgpack-python
 
     # Testing

devtools/conda-envs/psi.yaml

@@ -16,7 +16,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental=0.24.0
-  - pydantic=1.8.2  # test minimun stated version.
+  - pydantic>=2.1  # test minimum stated version.
+  - pydantic-settings
   - msgpack-python
 
     # Testing

devtools/conda-envs/qcore.yaml

@@ -11,7 +11,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.24
-  - pydantic >=1.8.2
+  - pydantic>=2.1
+  - pydantic-settings
   - tbb<2021
 
     # Testing

devtools/conda-envs/rdkit.yaml

@@ -11,7 +11,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.12.0
-  - pydantic>=1.0.0
+  - pydantic>=2.1
+  - pydantic-settings
 
     # Testing
   - pytest

devtools/conda-envs/torchani.yaml

@@ -11,7 +11,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.12.0
-  - pydantic>=1.0.0
+  - pydantic>=2.1
+  - pydantic-settings
 
   - pytorch
 

devtools/conda-envs/xtb.yaml

@@ -11,7 +11,8 @@ dependencies:
   - py-cpuinfo
   - psutil
   - qcelemental >=0.11.1
-  - pydantic >=1.8.2
+  - pydantic>=2.1
+  - pydantic-settings
 
     # Extras
   - gcp-correction

qcengine/compute.py

@@ -13,10 +13,7 @@
 from .util import compute_wrapper, environ_context, handle_output_metadata, model_wrapper
 
 if TYPE_CHECKING:
-    try:
-        from pydantic.v1.main import BaseModel
-    except ImportError:
-        from pydantic.main import BaseModel
+    from pydantic.main import BaseModel
     from qcelemental.models import AtomicResult
 
 

qcengine/config.py

@@ -9,10 +9,8 @@
 import socket
 from typing import Any, Dict, Optional, Union
 
-try:
-    import pydantic.v1 as pydantic
-except ImportError:
-    import pydantic
+from pydantic import BaseModel, ConfigDict, Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
 
 from .extras import get_information
 
@@ -64,7 +62,7 @@ def get_global(key: Optional[str] = None) -> Union[str, Dict[str, Any]]:
         return _global_values[key]
 
 
-class NodeDescriptor(pydantic.BaseModel):
+class NodeDescriptor(BaseModel):
     """
     Description of an individual node
     """
@@ -78,7 +76,7 @@ class NodeDescriptor(pydantic.BaseModel):
     memory_safety_factor: int = 10  # Percentage of memory as a safety factor
 
     # Specifications
-    ncores: Optional[int] = pydantic.Field(
+    ncores: Optional[int] = Field(
         None,
         description="""Number of cores accessible to each task on this node
     
@@ -88,7 +86,7 @@ class NodeDescriptor(pydantic.BaseModel):
     retries: int = 0
 
     # Cluster options
-    is_batch_node: bool = pydantic.Field(
+    is_batch_node: bool = Field(
         False,
         help="""Whether the node running QCEngine is a batch node
     
@@ -103,7 +101,7 @@ class NodeDescriptor(pydantic.BaseModel):
     ``mpiexec_command`` must always be used even for serial jobs (e.g., getting the version number)
     """,
     )
-    mpiexec_command: Optional[str] = pydantic.Field(
+    mpiexec_command: Optional[str] = Field(
         None,
         description="""Invocation for launching node-parallel tasks with MPI
         
@@ -140,31 +138,40 @@ def __init__(self, **data: Dict[str, Any]):
             if "{ranks_per_node}" not in self.mpiexec_command:
                 raise ValueError("mpiexec_command must explicitly state the number of ranks per node")
 
-    class Config:
-        extra = "forbid"
+    model_config = ConfigDict(
+        extra="forbid",
+    )
+
+
+#   class Config:
+#        extra = "forbid"
+#        allow_mutation = False
+#        json_encoders = {np.ndarray: lambda v: v.flatten().tolist(), complex: lambda v: (v.real, v.imag)}
+# ---
+#    model_config = ConfigDict(
+#        extra="forbid",
+#        frozen=True,
+#    )
 
 
-class TaskConfig(pydantic.BaseSettings):
+class TaskConfig(BaseSettings):
     """Description of the configuration used to launch a task."""
 
     # Specifications
-    ncores: int = pydantic.Field(None, description="Number cores per task on each node")
-    nnodes: int = pydantic.Field(None, description="Number of nodes per task")
-    memory: float = pydantic.Field(
-        None, description="Amount of memory in GiB (2^30 bytes; not GB = 10^9 bytes) per node."
-    )
+    ncores: int = Field(None, description="Number cores per task on each node")
+    nnodes: int = Field(None, description="Number of nodes per task")
+    memory: float = Field(None, description="Amount of memory in GiB (2^30 bytes; not GB = 10^9 bytes) per node.")
     scratch_directory: Optional[str]  # What location to use as scratch
     retries: int  # Number of retries on random failures
     mpiexec_command: Optional[str]  # Command used to launch MPI tasks, see NodeDescriptor
     use_mpiexec: bool = False  # Whether it is necessary to use MPI to run an executable
-    cores_per_rank: int = pydantic.Field(1, description="Number of cores per MPI rank")
-    scratch_messy: bool = pydantic.Field(
-        False, description="Leave scratch directory and contents on disk after completion."
-    )
+    cores_per_rank: int = Field(1, description="Number of cores per MPI rank")
+    scratch_messy: bool = Field(False, description="Leave scratch directory and contents on disk after completion.")
 
-    class Config(pydantic.BaseSettings.Config):
-        extra = "forbid"
-        env_prefix = "QCENGINE_"
+    model_config = SettingsConfigDict(
+        extra="forbid",
+        env_prefix="QCENGINE_",
+    )
 
 
 def _load_defaults() -> None:

qcengine/procedures/berny.py

@@ -3,7 +3,7 @@
 import sys
 import traceback
 from io import StringIO
-from typing import Any, Dict, Union
+from typing import Any, ClassVar, Dict, Union
 
 import numpy as np
 from qcelemental.models import OptimizationInput, OptimizationResult, FailedOperation
@@ -16,7 +16,7 @@
 
 
 class BernyProcedure(ProcedureHarness):
-    _defaults = {"name": "Berny", "procedure": "optimization"}
+    _defaults: ClassVar[Dict[str, Any]] = {"name": "Berny", "procedure": "optimization"}
 
     def found(self, raise_error: bool = False) -> bool:
         return which_import(

qcengine/procedures/geometric.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Union
+from typing import Any, ClassVar, Dict, Union
 
 from qcelemental.models import OptimizationInput, OptimizationResult
 from qcelemental.util import safe_version, which_import
@@ -8,13 +8,10 @@
 
 class GeometricProcedure(ProcedureHarness):
 
-    _defaults = {"name": "geomeTRIC", "procedure": "optimization"}
+    _defaults: ClassVar[Dict[str, Any]] = {"name": "geomeTRIC", "procedure": "optimization"}
 
     version_cache: Dict[str, str] = {}
 
-    class Config(ProcedureHarness.Config):
-        pass
-
     def found(self, raise_error: bool = False) -> bool:
         return which_import(
             "geometric",

qcengine/procedures/model.py

@@ -1,10 +1,7 @@
 import abc
 from typing import Any, Dict, Union
 
-try:
-    from pydantic.v1 import BaseModel
-except ImportError:
-    from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 from ..util import model_wrapper
 
@@ -14,9 +11,10 @@ class ProcedureHarness(BaseModel, abc.ABC):
     name: str
     procedure: str
 
-    class Config:
-        allow_mutation: False
-        extra: "forbid"
+    model_config = ConfigDict(
+        frozen=True,
+        extra="forbid",
+    )
 
     def __init__(self, **kwargs):
         super().__init__(**{**self._defaults, **kwargs})

qcengine/procedures/nwchem_opt/__init__.py

@@ -1,4 +1,4 @@
-from typing import Union, Dict, Any
+from typing import Any, ClassVar, Dict, Union
 
 from qcelemental.models import OptimizationInput, AtomicInput, OptimizationResult, Provenance
 
@@ -12,10 +12,7 @@
 class NWChemDriverProcedure(ProcedureHarness):
     """Structural relaxation using NWChem's optimizer"""
 
-    _defaults = {"name": "NWChemDriver", "procedure": "optimization"}
-
-    class Config(ProcedureHarness.Config):
-        pass
+    _defaults: ClassVar[Dict[str, Any]] = {"name": "NWChemDriver", "procedure": "optimization"}
 
     def found(self, raise_error: bool = False) -> bool:
         nwc_harness = NWChemHarness()

qcengine/procedures/optking.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Union
+from typing import Any, ClassVar, Dict, Union
 
 from qcelemental.models import OptimizationInput, OptimizationResult
 from qcelemental.util import safe_version, which_import
@@ -8,13 +8,10 @@
 
 class OptKingProcedure(ProcedureHarness):
 
-    _defaults = {"name": "OptKing", "procedure": "optimization"}
+    _defaults: ClassVar[Dict[str, Any]] = {"name": "OptKing", "procedure": "optimization"}
 
     version_cache: Dict[str, str] = {}
 
-    class Config(ProcedureHarness.Config):
-        pass
-
     def found(self, raise_error: bool = False) -> bool:
         return which_import(
             "optking",

qcengine/procedures/torsiondrive.py

@@ -1,7 +1,7 @@
 import io
 from collections import defaultdict
 from contextlib import redirect_stderr, redirect_stdout
-from typing import TYPE_CHECKING, Any, Dict, List, Tuple, Union
+from typing import TYPE_CHECKING, Any, ClassVar, Dict, List, Tuple, Union
 
 import numpy as np
 from qcelemental.models import FailedOperation, Molecule
@@ -16,10 +16,7 @@
 
 class TorsionDriveProcedure(ProcedureHarness):
 
-    _defaults = {"name": "TorsionDrive", "procedure": "torsiondrive"}
-
-    class Config(ProcedureHarness.Config):
-        pass
+    _defaults: ClassVar[Dict[str, Any]] = {"name": "TorsionDrive", "procedure": "torsiondrive"}
 
     def found(self, raise_error: bool = False) -> bool:
         return which_import(