diff --git a/openff/qcsubmit/datasets/datasets.py b/openff/qcsubmit/datasets/datasets.py index 75190c6c..e601d9c5 100644 --- a/openff/qcsubmit/datasets/datasets.py +++ b/openff/qcsubmit/datasets/datasets.py @@ -26,6 +26,7 @@ from qcportal.datasets.singlepoint import SinglepointDatasetNewEntry from qcportal.datasets.torsiondrive import TorsiondriveDatasetNewEntry from qcportal.records.singlepoint import SinglepointDriver, QCSpecification +from qcportal.records.optimization import OptimizationSpecification from typing_extensions import Literal from openff.qcsubmit.common_structures import ( @@ -53,7 +54,6 @@ if TYPE_CHECKING: from openff.toolkit.typing.engines.smirnoff import ForceField - from qcportal.records.optimization import OptimizationSpecification C = TypeVar("C", bound="Collection") E = TypeVar("E", bound=DatasetEntry)