fixed lint

Author: qcdyx

api/src/scripts/populate_db_gtfs.py

@@ -4,7 +4,6 @@
 
 import pycountry
 import pytz
-from sqlalchemy import func
 
 from scripts.load_dataset_on_create import publish_all
 from scripts.populate_db import DatabasePopulateHelper, set_up_configs
@@ -15,7 +14,6 @@
     Gtfsrealtimefeed,
     Location,
     Redirectingid,
-    Gtfsfeed,
 )
 from utils.data_utils import set_up_defaults
 
@@ -142,7 +140,7 @@ def process_feed_references(self, session: "Session"):
                 already_referenced_ids = {ref.id for ref in gtfs_feed.gtfs_rt_feeds}
                 if gtfs_feed and gtfs_rt_feed.id not in already_referenced_ids:
                     gtfs_rt_feed.gtfs_feeds = [gtfs_feed]
-                    #gtfs_feed.gtfs_rt_feeds.append(gtfs_rt_feed)                    # Flush to avoid FK violation
+                    # gtfs_feed.gtfs_rt_feeds.append(gtfs_rt_feed)                    # Flush to avoid FK violation
                     session.flush()
 
     def process_redirects(self, session: "Session"):

api/src/shared/db_models/gtfs_rt_feed_impl.py

@@ -1,7 +1,6 @@
 import logging
 
 from sqlalchemy.orm import Session
-from sqlalchemy import func
 
 from shared.database.database import with_db_session
 from shared.db_models.feed_impl import FeedImpl

api/tests/integration/populate_tests/test_populate.py

@@ -54,6 +54,7 @@ def test_is_official_overwrite(client: TestClient, values):
     json_response = response.json()
     assert json_response["official"] is expected_official, values["assert_fail_message"]
 
+
 def test_is_feed_reference_overwrite(client: TestClient):
     feed_id = "mdb-1562"
     response = client.request(
@@ -71,9 +72,4 @@ def test_is_feed_reference_overwrite(client: TestClient):
         headers=authHeaders,
     )
     json_response = response.json()
-    assert json_response["feed_references"] == ["mdb-50"]
-
-
-
-
-
+    assert json_response["feed_references"] == ["mdb-50"]