the rest

Author: loriab

qcelemental/models/basis.py

@@ -15,3 +15,5 @@
 BasisSet = qcelemental.models.v1.BasisSet
 ECPPotential = qcelemental.models.v1.ECPPotential
 ElectronShell = qcelemental.models.v1.ElectronShell
+HarmonicType = qcelemental.models.v1.HarmonicType
+ECPType = qcelemental.models.v1.ECPType

qcelemental/models/procedures.py

@@ -19,6 +19,7 @@
 TorsionDriveInput = qcelemental.models.v1.TorsionDriveInput
 TorsionDriveResult = qcelemental.models.v1.TorsionDriveResult
 OptimizationSpecification = qcelemental.models.v1.OptimizationSpecification
+TrajectoryProtocolEnum = qcelemental.models.v1.TrajectoryProtocolEnum
 
 # needed by QCFractal
 Model = qcelemental.models.v1.Model

qcelemental/models/results.py

@@ -17,8 +17,8 @@
 AtomicResultProtocols = qcelemental.models.v1.AtomicResultProtocols
 WavefunctionProperties = qcelemental.models.v1.WavefunctionProperties
 ErrorCorrectionProtocol = qcelemental.models.v1.ErrorCorrectionProtocol
-# WavefunctionProtocolEnum = qcelemental.models.v1.WavefunctionProtocolEnum
-# NativeFilesProtocolEnum = qcelemental.models.v1.NativeFilesProtocolEnum
+WavefunctionProtocolEnum = qcelemental.models.v1.WavefunctionProtocolEnum
+NativeFilesProtocolEnum = qcelemental.models.v1.NativeFilesProtocolEnum
 
 # needed by QCFractal
 Provenance = qcelemental.models.v1.Provenance

qcelemental/models/v1/__init__.py

@@ -60,6 +60,8 @@ def __repr__(self):
     "BasisSet",
     "ECPPotential",
     "ElectronShell",
+    "ECPType",
+    "HarmonicType",
     "ComputeError",
     "DriverEnum",
     "FailedOperation",
@@ -74,6 +76,7 @@ def __repr__(self):
     "OptimizationSpecification",
     "QCInputSpecification",
     "TDKeywords",
+    "TrajectoryProtocolEnum",
     "TorsionDriveInput",
     "TorsionDriveResult",
     "Result",
@@ -112,6 +115,8 @@ def _use_real_if_possible():
         "BasisSet": (".basis", "BasisSet"),
         "ECPPotential": (".basis", "ECPPotential"),
         "ElectronShell": (".basis", "ElectronShell"),
+        "ECPType": (".basis", "ECPType"),
+        "HarmonicType": (".basis", "HarmonicType"),
         "ComputeError": (".common_models", "ComputeError"),
         "DriverEnum": (".common_models", "DriverEnum"),
         "FailedOperation": (".common_models", "FailedOperation"),
@@ -126,6 +131,7 @@ def _use_real_if_possible():
         "OptimizationSpecification": (".procedures", "OptimizationSpecification"),
         "QCInputSpecification": (".procedures", "QCInputSpecification"),
         "TDKeywords": (".procedures", "TDKeywords"),
+        "TrajectoryProtocolEnum": (".procedures", "TrajectoryProtocolEnum"),
         "TorsionDriveInput": (".procedures", "TorsionDriveInput"),
         "TorsionDriveResult": (".procedures", "TorsionDriveResult"),
         "Result": (".results", "Result"),