Fix ValidationInfo import

Matgenix · Nov 19, 2023 · 0ecf1b5 · 0ecf1b5
1 parent 66cf919
commit 0ecf1b5
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/jobflow_remote/config/base.py b/src/jobflow_remote/config/base.py
@@ -3,10 +3,10 @@
 import traceback
 from enum import Enum
 from pathlib import Path
-from typing import TYPE_CHECKING, Annotated, Literal, Optional, Union
+from typing import Annotated, Literal, Optional, Union
 
 from jobflow import JobStore
-from pydantic import BaseModel, ConfigDict, Field, field_validator
+from pydantic import BaseModel, ConfigDict, Field, ValidationInfo, field_validator
 from qtoolkit.io import BaseSchedulerIO, scheduler_mapping
 
 from jobflow_remote.fireworks.launchpad import RemoteLaunchPad
@@ -15,9 +15,6 @@
 
 DEFAULT_JOBSTORE = {"docs_store": {"type": "MemoryStore"}}
 
-if TYPE_CHECKING:
-    from pydantic import ValidationInfo
-
 
 class RunnerOptions(BaseModel):
     """