tests.flow all pass. Now everything is GREEN ✅

2 files changed, 4 insertions, 2 deletions

diff --git a/jb/managers/assignment.py b/jb/managers/assignment.py
index c0a168e..6104bc5 100644
--- a/jb/managers/assignment.py
+++ b/jb/managers/assignment.py
@@ -2,7 +2,7 @@ from datetime import datetime, timezone
 from typing import Optional
 
 from psycopg import sql
-from pydantic import NonNegativeInt
+from pydantic import NonNegativeInt, PositiveInt
 
 from jb.managers import PostgresManager
 from jb.models.assignment import AssignmentStub, Assignment
@@ -215,7 +215,7 @@ class AssignmentManager(PostgresManager):
         return None
 
     def missing_tsid_count(
-        self, amt_worker_id: str, lookback_hrs: int = 24
+        self, amt_worker_id: str, lookback_hrs: PositiveInt = 24
     ) -> NonNegativeInt:
         """
         Look at this user's previous N hrs of submitted assignments.
diff --git a/jb/managers/thl.py b/jb/managers/thl.py
index 83f49f6..44b6496 100644
--- a/jb/managers/thl.py
+++ b/jb/managers/thl.py
@@ -23,6 +23,7 @@ import requests
 def get_user_profile(amt_worker_id: str) -> UserProfile:
     url = f"{settings.fsb_host}{settings.product_id}/user/{amt_worker_id}/profile/"
     res = requests.get(url).json()
+
     if res.get("detail") == "user not found":
         raise ValueError("user not found")
 
@@ -39,6 +40,7 @@ def get_user_blocked_or_not_exists(amt_worker_id: str) -> Optional[bool]:
     try:
         res = get_user_profile(amt_worker_id=amt_worker_id)
         return res.user.blocked if res.user.blocked is not None else False
+
     except ValueError as e:
         if e.args[0] == "user not found":
             return True