REFAC: Remove 'polars' dependency and update DataFrame usage to 'pand…

…as' in CLI to make the package lighter
pablofueros · Jan 29, 2025 · e5dc9ef · e5dc9ef
1 parent 658a535
commit e5dc9ef
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 19 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,7 +10,6 @@ requires-python = ">=3.12"
 dependencies = [
     "pandas>=2.2.3",
     "plotly>=5.24.1",
-    "polars>=1.20.0",
     "pydantic-xml>=2.14.1",
     "rich>=13.9.4",
     "typer>=0.15.1",

diff --git a/src/bbrpy/cli.py b/src/bbrpy/cli.py
@@ -1,8 +1,8 @@
 import pathlib
 import webbrowser
 
+import pandas as pd
 import plotly.express as px
-import polars as pl
 import rich
 import typer
 
@@ -51,7 +51,7 @@ def generate(
 
     # Generate the battery report and extract the capacity history
     battery_report = BatteryReport.generate()
-    history_df = pl.DataFrame([entry.model_dump() for entry in battery_report.History])
+    history_df = pd.DataFrame([entry.model_dump() for entry in battery_report.History])
 
     # Generate the capacity history visualization
     fig = px.line(

diff --git a/uv.lock b/uv.lock