mypy: Added annotatios for populate_db.py

zulip · Dec 8, 2016 · 9fec7ba · 9fec7ba
1 parent cf10dbb
commit 9fec7ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/zilencer/management/commands/populate_db.py b/zilencer/management/commands/populate_db.py
@@ -179,13 +179,13 @@ def handle(self, **options):
                                  Recipient.objects.filter(type=Recipient.STREAM)]
 
         # Extract a list of all users
-        user_profiles = list(UserProfile.objects.all()) # type: List[Any]
+        user_profiles = list(UserProfile.objects.all()) # type: List[UserProfile]
 
         if not options["test_suite"]:
             # Populate users with some bar data
             for user in user_profiles:
                 status = 1 # type: int
-                date = datetime.datetime.today() # type: datetime
+                date = datetime.datetime.today() # type: datetime.datetime
                 client = get_client("website")
                 for i in range(3):
                     client = get_client("API")