Extensive use of type checking. Movement of pytest conf towards handling managers (for db agnostic unittest). Starting to organize pytests.

1 files changed, 4 insertions, 4 deletions

diff --git a/jb/managers/assignment.py b/jb/managers/assignment.py
index fca72e8..dd3c866 100644
--- a/jb/managers/assignment.py
+++ b/jb/managers/assignment.py
@@ -28,7 +28,7 @@ class AssignmentManager(PostgresManager):
         with self.pg_config.make_connection() as conn:
             with conn.cursor() as c:
                 c.execute(query, data)
-                pk = c.fetchone()["id"]
+                pk = c.fetchone()["id"]  # type: ignore
                 conn.commit()
         stub.id = pk
         return None
@@ -62,7 +62,7 @@ class AssignmentManager(PostgresManager):
         with self.pg_config.make_connection() as conn:
             with conn.cursor() as c:
                 c.execute(query, data)
-                pk = c.fetchone()["id"]
+                pk = c.fetchone()["id"]  # type: ignore
                 conn.commit()
         assignment.id = pk
         return None
@@ -233,7 +233,7 @@ class AssignmentManager(PostgresManager):
                 "lookback_interval": f"{lookback_hrs} hour",
             },
         )
-        return int(res[0]["c"])
+        return int(res[0]["c"])  # type: ignore
 
     def rejected_count(
         self, amt_worker_id: str, lookback_hrs: int = 24
@@ -256,4 +256,4 @@ class AssignmentManager(PostgresManager):
                 "status": AssignmentStatus.Rejected.value,
             },
         )
-        return int(res[0]["c"])
+        return int(res[0]["c"])  # type: ignore