switch cache file to parquet

xoolive · Dec 13, 2024 · 2438f32 · 2438f32
1 parent 0dcee79
commit 2438f32
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/traffic/data/basic/aircraft.py b/src/traffic/data/basic/aircraft.py
@@ -247,14 +247,14 @@ def download_opensky(self) -> None:  # coverage: ignore
             engine="c",
             keep_default_na=False,
         )
-        self.data.to_pickle(self.cache_dir / "opensky_db.pkl")
+        self.data.to_parquet(self.cache_dir / "opensky_db.parquet")
 
     @property
     def opensky_db(self) -> pd.DataFrame:
-        if not (self.cache_dir / "opensky_db.pkl").exists():
+        if not (self.cache_dir / "opensky_db.parquet").exists():
             self.download_opensky()
         _log.info("Loading OpenSky aircraft database")
-        return pd.read_pickle(self.cache_dir / "opensky_db.pkl")
+        return pd.read_parquet(self.cache_dir / "opensky_db.parquet")
 
     def __getitem__(self: T, name: str | list[str]) -> None | T:
         """Requests an aircraft by icao24 or registration (exact match)."""