Fix some mypy types

Author: pamelafox

.vscode/settings.json

@@ -1,22 +1,4 @@
 {
-    "pgsql.connections": [
-        {
-            "id": "92CC1089-BAD0-44A4-B071-A50A6EC12B67",
-            "groupId": "F3347CD6-9995-4EE9-8D98-D88DB010FA5B",
-            "authenticationType": "SqlLogin",
-            "connectTimeout": 15,
-            "applicationName": "vscode-pgsql",
-            "clientEncoding": "utf8",
-            "sslmode": "prefer",
-            "server": "localhost",
-            "user": "admin",
-            "password": "",
-            "savePassword": true,
-            "database": "postgres",
-            "profileName": "local-pg",
-            "expiresOn": 0
-        }
-    ],
     "python.testing.pytestArgs": [
         "tests"
     ],

convert_csv_json.py

@@ -1,6 +1,7 @@
 import ast
 import csv
 import json
+from typing import Any
 
 # Read CSV file - Using the correct dialect to handle quotes properly
 with open("pittsburgh_restaurants.csv", encoding="utf-8") as csv_file:
@@ -15,7 +16,7 @@
     item = {}
     for i in range(len(header)):
         if i < len(row):  # Ensure we don't go out of bounds
-            value = row[i].strip()
+            value: Any = row[i].strip()
             # Check if the value looks like a JSON array
             if value.startswith("[") and value.endswith("]"):
                 try:

tests/data.py

@@ -10,7 +10,7 @@ class TestData(ItemPublic):
     name="Balvanera",
     cuisine="Argentine, Wine Bars, Steakhouses",
     rating=4.5,
-    price_level=2.0,
+    price_level=2,
     review_count=132,
     description="Balvanera is an Argentine restaurant that also serves as a wine bar and steakhouse.",
     menu_summary="The cuisine is Argentine, likely featuring grilled meats and steaks, along with a selection of wines.",  # noqa: E501