feat: add GTFS Loader

.flake8

@@ -1,7 +1,7 @@
 [flake8]
 max-line-length = 100
 max-doc-length = 100
-extend-ignore = E203
+extend-ignore = E203,B028
 exclude =
     .git,
     .venv,

CHANGELOG.md

@@ -7,6 +7,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [Unreleased] - 2022-MM-DD
 
 ### Added
+- GTFS Loader from gtfs2vec paper
 
 ### Changed
 - Change embedders and joiners interface to have `.transform` method

examples/loaders/files/.gitignore

@@ -0,0 +1,2 @@
+# example GTFS used in notebook
+example.zip

examples/loaders/gtfs_loader.ipynb

@@ -0,0 +1,132 @@
+{
+ "cells": [
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# GTFS Loader Example"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from pathlib import Path\n",
+    "from srai.loaders import GTFSLoader\n",
+    "import gtfs_kit as gk\n",
+    "import geopandas as gpd\n",
+    "import numpy as np\n",
+    "from shapely.geometry import Point\n",
+    "from srai.utils.constants import WGS84_CRS"
+   ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Download an example GTFS feed from Wroclaw, Poland\n",
+    "\n",
+    "In this notebook we use the GTFS feed for Wroclaw, Poland as an example, which is available in Wroclaw's open data repository[1]. This download uses transitfeeds.com[2] to download the feed, but you can also download the feed directly from the Wroclaw open data repository.\n",
+    "\n",
+    "1. https://www.wroclaw.pl/open-data/dataset/rozkladjazdytransportupublicznegoplik_data\n",
+    "2. https://transitfeeds.com/p/mpk-wroc-aw/663"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "! wget -O files/example.zip https://transitfeeds.com/p/mpk-wroc-aw/663/20221221/download"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "wroclaw_gtfs = Path().resolve() / \"files\" / \"example.zip\""
+   ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Peek at the feed using `gtfs_kit` directly"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "feed = gk.read_feed(wroclaw_gtfs, dist_units=\"km\")\n",
+    "\n",
+    "stops_df = feed.stops[[\"stop_id\", \"stop_lat\", \"stop_lon\"]].set_index(\"stop_id\")\n",
+    "stops_df[\"geometry\"] = stops_df.apply(lambda row: Point(row[\"stop_lon\"], row[\"stop_lat\"]), axis=1)\n",
+    "\n",
+    "stops_gdf = gpd.GeoDataFrame(\n",
+    "    stops_df,\n",
+    "    geometry=\"geometry\",\n",
+    "    crs=WGS84_CRS,\n",
+    ")\n",
+    "\n",
+    "stops_gdf.plot(markersize=1)"
+   ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Use GTFSLoader to load stops statistics from the feed"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gtfs_loader = GTFSLoader()\n",
+    "trips_gdf = gtfs_loader.load(wroclaw_gtfs)\n",
+    "\n",
+    "print(trips_gdf.columns)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.8.14"
+  },
+  "vscode": {
+   "interpreter": {
+    "hash": "2102df1d74c890a541339243f341c4aa4ffb38ff8671f11b79e33b00adbba0dd"
+   }
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}