FIX: Fix pandas 3.0 issues

Author: nmacholl

databento/common/dbnstore.py

@@ -1528,8 +1528,7 @@ def _format_px(
 
         if price_type == PriceType.DECIMAL:
             df[px_fields] = (
-                df[px_fields].replace(UNDEF_PRICE, np.nan).applymap(decimal.Decimal)
-                / FIXED_PRICE_SCALE
+                df[px_fields].replace(UNDEF_PRICE, np.nan).map(decimal.Decimal) / FIXED_PRICE_SCALE
             )
         elif price_type == PriceType.FLOAT:
             df[px_fields] = df[px_fields].replace(UNDEF_PRICE, np.nan) / FIXED_PRICE_SCALE

pyproject.toml

@@ -51,10 +51,7 @@ pytest = "^7.4.2"
 pytest-asyncio = "==0.21.1"
 ruff = "^0.0.291"
 types-requests = "^2.30.0.0"
-tomli = "^2.0.1"
 teamcity-messages = "^1.32"
-types-pytz = "^2024.1.0.20240203"
-types-aiofiles = "^23.2.0.20240403"
 
 [build-system]
 requires = ["poetry-core"]

tests/test_historical_bento.py

@@ -8,11 +8,11 @@
 from typing import Any
 from typing import Literal
 from unittest.mock import MagicMock
+from zoneinfo import ZoneInfo
 
 import numpy as np
 import pandas as pd
 import pytest
-import pytz
 import zstandard
 from databento_dbn import Compression
 from databento_dbn import DBNRecord
@@ -1579,24 +1579,24 @@ def test_dbnstore_to_df_with_timezone(
 @pytest.mark.parametrize(
     "timezone",
     [
-        pytz.timezone("US/Central"),
-        pytz.timezone("US/Eastern"),
-        pytz.timezone("Europe/Vienna"),
-        pytz.timezone("Asia/Dubai"),
-        pytz.timezone("UTC"),
+        ZoneInfo("US/Central"),
+        ZoneInfo("US/Eastern"),
+        ZoneInfo("Europe/Vienna"),
+        ZoneInfo("Asia/Dubai"),
+        ZoneInfo("UTC"),
     ],
 )
 @pytest.mark.parametrize(
     "schema",
     [pytest.param(schema, id=str(schema)) for schema in Schema.variants()],
 )
-def test_dbnstore_to_df_with_pytz_timezone(
+def test_dbnstore_to_df_with_zoneinfo(
     test_data_path: Callable[[Dataset, Schema], Path],
     schema: Schema,
-    timezone: pytz.BaseTzInfo,
+    timezone: ZoneInfo,
 ) -> None:
     """
-    Test that setting the `tz` parameter in `DBNStore.to_df` accepts `pytz`
+    Test that setting the `tz` parameter in `DBNStore.to_df` accepts `ZoneInfo`
     timezone objects.
     """
     # Arrange

tests/test_release.py

@@ -7,7 +7,7 @@
 from datetime import date
 
 import pytest
-import tomli
+import tomllib
 
 import databento
 from tests import PROJECT_ROOT
@@ -43,7 +43,7 @@ def fixture_pyproject_version() -> str:
     """
     # Arrange, Act, Assert
     with open(PROJECT_ROOT / "pyproject.toml", "rb") as pyproject:
-        data = tomli.load(pyproject)
+        data = tomllib.load(pyproject)
     return data["project"]["version"]