diff options
Diffstat (limited to 'src/sec_certs/utils')
| -rw-r--r-- | src/sec_certs/utils/helpers.py | 3 | ||||
| -rw-r--r-- | src/sec_certs/utils/parallel_processing.py | 3 | ||||
| -rw-r--r-- | src/sec_certs/utils/plot_utils.py | 10 | ||||
| -rw-r--r-- | src/sec_certs/utils/sanitization.py | 2 |
4 files changed, 9 insertions, 9 deletions
diff --git a/src/sec_certs/utils/helpers.py b/src/sec_certs/utils/helpers.py index c103a2d3..c0440008 100644 --- a/src/sec_certs/utils/helpers.py +++ b/src/sec_certs/utils/helpers.py @@ -4,11 +4,12 @@ import hashlib import logging import re import time +from collections.abc import Collection from contextlib import nullcontext from datetime import datetime from functools import partial from pathlib import Path -from typing import Any, Collection +from typing import Any import numpy as np import pkgconfig diff --git a/src/sec_certs/utils/parallel_processing.py b/src/sec_certs/utils/parallel_processing.py index ae3f0b44..a30e080b 100644 --- a/src/sec_certs/utils/parallel_processing.py +++ b/src/sec_certs/utils/parallel_processing.py @@ -1,9 +1,10 @@ from __future__ import annotations import time +from collections.abc import Callable, Iterable from multiprocessing import cpu_count from multiprocessing.pool import Pool, ThreadPool -from typing import Any, Callable, Iterable +from typing import Any from sec_certs.configuration import config from sec_certs.utils.tqdm import tqdm diff --git a/src/sec_certs/utils/plot_utils.py b/src/sec_certs/utils/plot_utils.py index b9e7ac7c..eca79070 100644 --- a/src/sec_certs/utils/plot_utils.py +++ b/src/sec_certs/utils/plot_utils.py @@ -1,5 +1,3 @@ -from typing import Dict, List, Tuple - from networkx import DiGraph from pandas import DataFrame @@ -20,8 +18,8 @@ def get_cert_property(df: DataFrame, cert_id: int, column: str) -> str: def get_fips_cert_references_graph( - df: DataFrame, cert_id: int, colour_mapper: Dict[str, str] -) -> Tuple[DiGraph, List[str]]: + df: DataFrame, cert_id: int, colour_mapper: dict[str, str] +) -> tuple[DiGraph, list[str]]: if cert_id not in df["cert_id"].unique(): raise ValueError(f"Cert ID: {cert_id} is not in the dataset") @@ -45,7 +43,7 @@ def get_fips_cert_references_graph( return graph, colour_map -def get_most_referenced_cert_graph(df: DataFrame, status_colour_mapper: Dict[str, str]) -> Tuple[DiGraph, List[str]]: +def get_most_referenced_cert_graph(df: DataFrame, status_colour_mapper: dict[str, str]) -> tuple[DiGraph, list[str]]: graph = DiGraph() colour_map = [] max_referenced_by_num = df["incoming_direct_references_count"].max() @@ -66,7 +64,7 @@ def get_most_referenced_cert_graph(df: DataFrame, status_colour_mapper: Dict[str return graph, colour_map -def get_most_referencing_cert_graph(df: DataFrame, status_colour_mapper: Dict[str, str]) -> Tuple[DiGraph, List[str]]: +def get_most_referencing_cert_graph(df: DataFrame, status_colour_mapper: dict[str, str]) -> tuple[DiGraph, list[str]]: graph = DiGraph() colour_map = [] max_referencing_num = df["outgoing_direct_references_count"].max() diff --git a/src/sec_certs/utils/sanitization.py b/src/sec_certs/utils/sanitization.py index 3563e1f4..17e34362 100644 --- a/src/sec_certs/utils/sanitization.py +++ b/src/sec_certs/utils/sanitization.py @@ -28,7 +28,7 @@ def sanitize_date(record: pd.Timestamp | date | np.datetime64) -> date | None: return None if isinstance(record, pd.Timestamp): return record.date() - if isinstance(record, (date, type(None))): + if isinstance(record, date | type(None)): return record raise ValueError("Unsupported type given as input") |
