fix import, tests, etc

docs/changelog.rst

@@ -26,6 +26,7 @@ Changelog
 Breaking Changes
 ++++++++++++++++
 * The very old model names `ResultInput`, `Result`, `ResultProperties`, `Optimization` deprecated in 2019 are now only available through `qcelelemental.models.v1`
+* ``models.v2`` do not support AutoDoc. The AutoDoc routines have been left at pydantic v1 syntax. Use autodoc-pydantic for Sphinx instead.
 
 New Features
 ++++++++++++

qcelemental/models/__init__.py

@@ -1 +1,10 @@
+try:
+    import pydantic
+except ImportError:  # pragma: no cover
+    raise ImportError(
+        "Python module pydantic not found. Solve by installing it: "
+        "`conda install pydantic -c conda-forge` or `pip install pydantic`"
+    )
+
+from . import v1, v2
 from .v1 import *

qcelemental/models/v1/__init__.py

@@ -1,11 +1,3 @@
-try:
-    import pydantic
-except ImportError:  # pragma: no cover
-    raise ImportError(
-        "Python module pydantic not found. Solve by installing it: "
-        "`conda install pydantic -c conda-forge` or `pip install pydantic`"
-    )
-
 from . import types
 from .align import AlignmentMill
 from .basemodels import AutodocBaseSettings  # remove when QCFractal merges `next`

qcelemental/models/v1/align.py

@@ -1,11 +1,7 @@
 from typing import Optional
 
 import numpy as np
-
-try:
-    from pydantic.v1 import Field, validator
-except ImportError:  # Will also trap ModuleNotFoundError
-    from pydantic import Field, validator
+from pydantic.v1 import Field, validator
 
 from ...util import blockwise_contract, blockwise_expand
 from .basemodels import ProtoModel

qcelemental/models/v1/basemodels.py

@@ -3,13 +3,8 @@
 from typing import Any, Dict, Optional, Set, Union
 
 import numpy as np
-
-try:
-    from pydantic.v1 import BaseSettings  # remove when QCFractal merges `next`
-    from pydantic.v1 import BaseModel
-except ImportError:  # Will also trap ModuleNotFoundError
-    from pydantic import BaseSettings  # remove when QCFractal merges `next`
-    from pydantic import BaseModel
+from pydantic.v1 import BaseSettings  # remove when QCFractal merges `next`
+from pydantic.v1 import BaseModel
 
 from qcelemental.util import deserialize, serialize
 from qcelemental.util.autodocs import AutoPydanticDocGenerator  # remove when QCFractal merges `next`

qcelemental/models/v1/basis.py

@@ -1,10 +1,7 @@
 from enum import Enum
 from typing import Dict, List, Optional
 
-try:
-    from pydantic.v1 import ConstrainedInt, Field, constr, validator
-except ImportError:  # Will also trap ModuleNotFoundError
-    from pydantic import ConstrainedInt, Field, constr, validator
+from pydantic.v1 import ConstrainedInt, Field, constr, validator
 
 from ...exceptions import ValidationError
 from .basemodels import ProtoModel, qcschema_draft

qcelemental/models/v1/common_models.py

@@ -2,20 +2,13 @@
 from typing import TYPE_CHECKING, Any, Dict, Optional, Union
 
 import numpy as np
-
-try:
-    from pydantic.v1 import Field
-except ImportError:  # Will also trap ModuleNotFoundError
-    from pydantic import Field
+from pydantic.v1 import Field
 
 from .basemodels import ProtoModel, qcschema_draft
 from .basis import BasisSet
 
 if TYPE_CHECKING:
-    try:
-        from pydantic.v1.typing import ReprArgs
-    except ImportError:  # Will also trap ModuleNotFoundError
-        from pydantic.typing import ReprArgs
+    from pydantic.v1.typing import ReprArgs
 
 
 # Encoders, to be deprecated

qcelemental/models/v1/molecule.py

@@ -10,11 +10,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Iterable, List, Optional, Tuple, Union, cast
 
 import numpy as np
-
-try:
-    from pydantic.v1 import ConstrainedFloat, ConstrainedInt, Field, constr, validator
-except ImportError:  # Will also trap ModuleNotFoundError
-    from pydantic import ConstrainedFloat, ConstrainedInt, Field, constr, validator
+from pydantic.v1 import ConstrainedFloat, ConstrainedInt, Field, constr, validator
 
 # molparse imports separated b/c https://github.com/python/mypy/issues/7203
 from ...molparse.from_arrays import from_arrays
@@ -31,10 +27,7 @@
 from .types import Array
 
 if TYPE_CHECKING:
-    try:
-        from pydantic.v1.typing import ReprArgs
-    except ImportError:  # Will also trap ModuleNotFoundError
-        from pydantic.typing import ReprArgs
+    from pydantic.v1.typing import ReprArgs
 
 # Rounding quantities for hashing
 GEOMETRY_NOISE = 8

qcelemental/models/v1/procedures.py

@@ -1,10 +1,7 @@
 from enum import Enum
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple
 
-try:
-    from pydantic.v1 import Field, conlist, constr, validator
-except ImportError:  # Will also trap ModuleNotFoundError
-    from pydantic import Field, conlist, constr, validator
+from pydantic.v1 import Field, conlist, constr, validator
 
 from ...util import provenance_stamp
 from .basemodels import ProtoModel
@@ -23,10 +20,7 @@
 from .results import AtomicResult
 
 if TYPE_CHECKING:
-    try:
-        from pydantic.v1.typing import ReprArgs
-    except ImportError:  # Will also trap ModuleNotFoundError
-        from pydantic.typing import ReprArgs
+    from pydantic.v1.typing import ReprArgs
 
 
 class TrajectoryProtocolEnum(str, Enum):

qcelemental/models/v1/results.py

@@ -3,11 +3,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Optional, Set, Union
 
 import numpy as np
-
-try:
-    from pydantic.v1 import Field, constr, validator
-except ImportError:  # Will also trap ModuleNotFoundError
-    from pydantic import Field, constr, validator
+from pydantic.v1 import Field, constr, validator
 
 from ...util import provenance_stamp
 from .basemodels import ProtoModel, qcschema_draft
@@ -17,10 +13,7 @@
 from .types import Array
 
 if TYPE_CHECKING:
-    try:
-        from pydantic.v1.typing import ReprArgs
-    except ImportError:  # Will also trap ModuleNotFoundError
-        from pydantic.typing import ReprArgs
+    from pydantic.v1.typing import ReprArgs
 
 
 class AtomicResultProperties(ProtoModel):

qcelemental/models/v2/__init__.py

@@ -0,0 +1,19 @@
+from . import types
+from .align import AlignmentMill
+from .basemodels import ProtoModel
+from .basis import BasisSet
+from .common_models import ComputeError, DriverEnum, FailedOperation, Provenance
+from .molecule import Molecule
+from .procedures import OptimizationInput, OptimizationResult
+from .results import AtomicInput, AtomicResult, AtomicResultProperties
+
+
+def qcschema_models():
+    return [
+        AtomicInput,
+        AtomicResult,
+        AtomicResultProperties,
+        BasisSet,
+        Molecule,
+        Provenance,
+    ]

qcelemental/models/v2/basemodels.py

@@ -5,10 +5,8 @@
 
 import numpy as np
 from pydantic import BaseModel, ConfigDict, model_serializer
-from pydantic_settings import BaseSettings  # remove when QCFractal merges `next`
 
 from qcelemental.util import deserialize, serialize
-from qcelemental.util.autodocs import AutoPydanticDocGenerator  # remove when QCFractal merges `next`
 
 
 def _repr(self) -> str:
@@ -279,10 +277,4 @@ def _merge_config_with(cls, *args, **kwargs):
         return ExtendedConfigDict(**output_dict)
 
 
-# remove when QCFractal merges `next`
-class AutodocBaseSettings(BaseSettings):
-    def __init_subclass__(cls) -> None:
-        cls.__doc__ = AutoPydanticDocGenerator(cls, always_apply=True)
-
-
 qcschema_draft = "http://json-schema.org/draft-04/schema#"

qcelemental/models/v2/procedures.py

@@ -261,18 +261,3 @@ class TorsionDriveResult(TorsionDriveInput):
     )
     error: Optional[ComputeError] = Field(None, description=str(ComputeError.__doc__))
     provenance: Provenance = Field(..., description=str(Provenance.__doc__))
-
-
-def Optimization(*args, **kwargs):
-    """QC Optimization Results Schema.
-
-    .. deprecated:: 0.12
-       Use :py:func:`qcelemental.models.OptimizationResult` instead.
-
-    """
-    from warnings import warn
-
-    warn(
-        "Optimization has been renamed to OptimizationResult and will be removed as soon as v0.13.0", DeprecationWarning
-    )
-    return OptimizationResult(*args, **kwargs)

qcelemental/models/v2/results.py

@@ -831,69 +831,3 @@ def _native_file_protocol(cls, value, info):
         for rk in return_keep:
             ret[rk] = files.get(rk, None)
         return ret
-
-
-class ResultProperties(AtomicResultProperties):
-    """QC Result Properties Schema.
-
-    .. deprecated:: 0.12
-       Use :py:func:`qcelemental.models.AtomicResultProperties` instead.
-
-    """
-
-    def __init__(self, *args, **kwargs):
-        from warnings import warn
-
-        warn(
-            "ResultProperties has been renamed to AtomicResultProperties and will be removed as soon as v0.13.0",
-            DeprecationWarning,
-        )
-        super().__init__(*args, **kwargs)
-
-
-class ResultProtocols(AtomicResultProtocols):
-    """QC Result Protocols Schema.
-
-    .. deprecated:: 0.12
-       Use :py:func:`qcelemental.models.AtomicResultProtocols` instead.
-
-    """
-
-    def __init__(self, *args, **kwargs):
-        from warnings import warn
-
-        warn(
-            "ResultProtocols has been renamed to AtomicResultProtocols and will be removed as soon as v0.13.0",
-            DeprecationWarning,
-        )
-        super().__init__(*args, **kwargs)
-
-
-class ResultInput(AtomicInput):
-    """QC Input Schema.
-
-    .. deprecated:: 0.12
-       Use :py:func:`qcelemental.models.AtomicInput` instead.
-
-    """
-
-    def __init__(self, *args, **kwargs):
-        from warnings import warn
-
-        warn("ResultInput has been renamed to AtomicInput and will be removed as soon as v0.13.0", DeprecationWarning)
-        super().__init__(*args, **kwargs)
-
-
-class Result(AtomicResult):
-    """QC Result Schema.
-
-    .. deprecated:: 0.12
-       Use :py:func:`qcelemental.models.AtomicResult` instead.
-
-    """
-
-    def __init__(self, *args, **kwargs):
-        from warnings import warn
-
-        warn("Result has been renamed to AtomicResult and will be removed as soon as v0.13.0", DeprecationWarning)
-        super().__init__(*args, **kwargs)

qcelemental/tests/addons.py

@@ -62,12 +62,13 @@ def xfail_on_pubchem_busy():
 
 
 def drop_qcsk(instance, tnm: str, schema_name: str = None):
-    if isinstance(instance, qcelemental.models.ProtoModel) and schema_name is None:
+    is_model = isinstance(instance, (qcelemental.models.v1.ProtoModel, qcelemental.models.v2.ProtoModel))
+    if is_model and schema_name is None:
         schema_name = type(instance).__name__
     drop = (_data_path / schema_name / tnm).with_suffix(".json")
 
     with open(drop, "w") as fp:
-        if isinstance(instance, qcelemental.models.ProtoModel):
+        if is_model:
             # fp.write(instance.json(exclude_unset=True, exclude_none=True))  # works but file is one-line
             instance = json.loads(instance.json(exclude_unset=True, exclude_none=True))
         elif isinstance(instance, dict):
@@ -83,7 +84,7 @@ def Molecule(request):
     if request.param == "v1":
         return qcelemental.models.v1.Molecule
     elif request.param == "v2":
-        return qcelemental.models.v1.Molecule  # TODO v2
+        return qcelemental.models.v2.Molecule
     else:
         return qcelemental.models.Molecule
 
@@ -93,6 +94,6 @@ def schema_versions(request):
     if request.param == "v1":
         return qcelemental.models.v1
     elif request.param == "v2":
-        return qcelemental.models.v1  # TODO v2
+        return qcelemental.models.v2
     else:
         return qcelemental.models