added check for TYPE_CHECKING

Author: pbashyal-nmdp

pyard/__init__.py

@@ -21,8 +21,10 @@
 #    > http://www.fsf.org/licensing/licenses/lgpl.html
 #    > http://www.opensource.org/licenses/lgpl-license.php
 #
-from .blender import blender as dr_blender
 from .constants import DEFAULT_CACHE_SIZE
+
+# exports for `pyard`
+from .blender import blender as dr_blender
 from .misc import get_imgt_db_versions as db_versions
 
 __author__ = """NMDP Bioinformatics"""

pyard/db.py

@@ -420,7 +420,7 @@ def find_xx_for_serology(connection: sqlite3.Connection, serology: str) -> str:
     :param serology: serology for which to find XX allele
     :return:  XX allele for given serology
     """
-    query = f"SELECT xx FROM serology_mapping WHERE serology = ?"
+    query = "SELECT xx FROM serology_mapping WHERE serology = ?"
     cursor = connection.execute(query, (serology,))
     results = cursor.fetchone()
     if results:
@@ -581,7 +581,7 @@ def save_serology_mappings(db_connection, sero_mapping):
     # Drop the table first
     cursor.execute("DROP TABLE IF EXISTS serology_mapping")
     # Create table
-    create_table_sql = f"""CREATE TABLE serology_mapping (
+    create_table_sql = """CREATE TABLE serology_mapping (
                             serology TEXT PRIMARY KEY,
                             allele_list TEXT,
                             lgx_allele_list TEXT,
@@ -592,7 +592,7 @@ def save_serology_mappings(db_connection, sero_mapping):
     rows = ((k, v[0], v[1], v[2]) for k, v in sero_mapping.items())
 
     # insert
-    cursor.executemany(f"INSERT INTO serology_mapping VALUES (?, ?, ?, ?)", rows)
+    cursor.executemany("INSERT INTO serology_mapping VALUES (?, ?, ?, ?)", rows)
 
     # commit transaction - writes to the db
     db_connection.commit()

pyard/reducers/base_reducer.py

@@ -2,6 +2,11 @@
 
 from abc import ABC, abstractmethod
 
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ..ard import ARD
+
 
 class Reducer(ABC):
     """Base class for all reduction strategies"""

pyard/reducers/reducer_factory.py

@@ -13,6 +13,11 @@
 from .w_reducer import WReducer
 from ..constants import VALID_REDUCTION_TYPE
 
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ..ard import ARD
+
 
 class StrategyFactory:
     """Factory for creating reduction strategies"""