Add annotations import everywhere.

6 files changed, 15 insertions, 2 deletions

diff --git a/src/sec_certs/utils/cc_html_parsing.py b/src/sec_certs/utils/cc_html_parsing.py
index 956353bf..2603b3ce 100644
--- a/src/sec_certs/utils/cc_html_parsing.py
+++ b/src/sec_certs/utils/cc_html_parsing.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from datetime import datetime
 from typing import Any
diff --git a/src/sec_certs/utils/label_studio_utils.py b/src/sec_certs/utils/label_studio_utils.py
index 9c70b938..82d663be 100644
--- a/src/sec_certs/utils/label_studio_utils.py
+++ b/src/sec_certs/utils/label_studio_utils.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 import logging
 from pathlib import Path
diff --git a/src/sec_certs/utils/nvd_dataset_builder.py b/src/sec_certs/utils/nvd_dataset_builder.py
index bffde677..eb62df15 100644
--- a/src/sec_certs/utils/nvd_dataset_builder.py
+++ b/src/sec_certs/utils/nvd_dataset_builder.py
@@ -5,6 +5,7 @@ import logging
 import math
 import random
 import time
+import typing
 from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
 from datetime import datetime
@@ -16,8 +17,10 @@ import requests
 from requests import RequestException, Response
 
 from sec_certs import constants
-from sec_certs.dataset.cpe import CPEDataset, CPEMatchDict
-from sec_certs.dataset.cve import CVEDataset
+
+if typing.TYPE_CHECKING:
+    from sec_certs.dataset.cpe import CPEDataset, CPEMatchDict
+    from sec_certs.dataset.cve import CVEDataset
 from sec_certs.utils.parallel_processing import process_parallel
 
 logger = logging.getLogger(__name__)
diff --git a/src/sec_certs/utils/plot_utils.py b/src/sec_certs/utils/plot_utils.py
index eca79070..f289cc06 100644
--- a/src/sec_certs/utils/plot_utils.py
+++ b/src/sec_certs/utils/plot_utils.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from networkx import DiGraph
 from pandas import DataFrame
 
diff --git a/src/sec_certs/utils/profiling.py b/src/sec_certs/utils/profiling.py
index 7da67070..4c7d0911 100644
--- a/src/sec_certs/utils/profiling.py
+++ b/src/sec_certs/utils/profiling.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import gc
 from contextlib import contextmanager
 from datetime import datetime
diff --git a/src/sec_certs/utils/tqdm.py b/src/sec_certs/utils/tqdm.py
index 0a23903f..c7c8958b 100644
--- a/src/sec_certs/utils/tqdm.py
+++ b/src/sec_certs/utils/tqdm.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from tqdm.auto import tqdm as tqdm_original
 
 from sec_certs.configuration import config