chore: Hopefully fixed problem with circular import

1 files changed, 5 insertions, 2 deletions

diff --git a/sec_certs/sample/common_criteria.py b/sec_certs/sample/common_criteria.py
index 8ae73eba..32855479 100644
--- a/sec_certs/sample/common_criteria.py
+++ b/sec_certs/sample/common_criteria.py
@@ -5,14 +5,13 @@ from datetime import date, datetime
 from enum import Enum
 from functools import partial
 from pathlib import Path
-from typing import Any, Callable, ClassVar, Dict, List, Optional, Set, Tuple, Union
+from typing import TYPE_CHECKING, Any, Callable, ClassVar, Dict, List, Optional, Set, Tuple, Union
 
 import requests
 from bs4 import Tag
 
 from sec_certs import constants as constants
 from sec_certs import helpers
-from sec_certs.dataset.common_criteria import CCDataset
 from sec_certs.model.cpe_matching import CPEClassifier
 from sec_certs.model.dependency_finder import References
 from sec_certs.sample.certificate import Certificate, Heuristics, logger
@@ -20,6 +19,10 @@ from sec_certs.sample.protection_profile import ProtectionProfile
 from sec_certs.serialization.json import ComplexSerializableType
 from sec_certs.serialization.pandas import PandasSerializableType
 
+if TYPE_CHECKING:
+    from sec_certs.dataset.common_criteria import CCDataset
+
+
 HEADERS = {
     "anssi": helpers.search_only_headers_anssi,
     "bsi": helpers.search_only_headers_bsi,