From 3b725c7354b185cc748f1d8ebfac6bab5e17b4aa Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Thu, 30 Nov 2023 09:58:23 +0300
Subject: Fix errors caught by pylint and mypy.

---
 scripts/insert_data.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'scripts/insert_data.py')

diff --git a/scripts/insert_data.py b/scripts/insert_data.py
index 45b6dd5..4dbf27c 100644
--- a/scripts/insert_data.py
+++ b/scripts/insert_data.py
@@ -66,6 +66,7 @@ def strains_info(
         return {strain["Name"]: strain for strain in cursor.fetchall()}
 
 def read_datavalues(filepath, headings, strain_info):
+    """Read numerical, data values from the file."""
     return {
         str(row["ProbeSetID"]): tuple({
             "ProbeSetName": str(row["ProbeSetID"]),
@@ -149,6 +150,7 @@ def insert_probesets(filepath: str,
                      platform_id: int,
                      headings: tuple[str, ...],
                      session_rand_str: str) -> tuple[str, ...]:
+    """Save new ProbeSets into the database."""
     probeset_query = (
         "INSERT INTO ProbeSet(ChipId, Name) "
         "VALUES (%(ChipId)s, %(Name)s) ")
@@ -157,7 +159,7 @@ def insert_probesets(filepath: str,
         "Name": f"{row['Name']}{session_rand_str}",
         "ChipId": platform_id
     } for row in read_probesets(filepath, headings))
-    probeset_names = tuple()
+    probeset_names: tuple[str, ...] = tuple()
     with dbconn.cursor(cursorclass=DictCursor) as cursor:
         while True:
             probeset_params = tuple(take(the_probesets, 10000))
@@ -209,7 +211,8 @@ def insert_means(# pylint: disable=[too-many-locals, too-many-arguments]
     #   which means that we cannot have 2 (or more) ProbeSets which share both
     #   the name and chip_id (platform) at the same time.
     rand_str = f"::RAND_{random_string()}"
-    pset_ids = {
+    pset_ids = {# pylint: disable=[unnecessary-comprehension]
+        # Look into simply doing dict(probeset_ids(...))
         name: pset_id
         for name, pset_id in probeset_ids(
                 dbconn,
-- 
cgit v1.2.3