diff options
Diffstat (limited to 'jb/managers/assignment.py')
| -rw-r--r-- | jb/managers/assignment.py | 8 |
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 |