Skip to content

Commit 749bd2a

Browse files
committed
Don't exclude retail
1 parent e919259 commit 749bd2a

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tariff_fetch/cli.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
from ._cli.types import Provider, StateCode, Utility
1818

1919
ENTITY_TYPES_SORTORDER = ["Investor Owned", "Cooperative", "Municipal"]
20-
ENTITY_TYPES_EXCLUDE = ["Retail Power Marketer"]
2120

2221

2322
def prompt_state() -> StateCode:
@@ -44,7 +43,6 @@ def prompt_utility(state: str) -> Utility:
4443
pl.read_parquet(CORE_EIA861_Yearly_Sales.https)
4544
.filter(pl.col("state") == state.upper())
4645
.filter(pl.col("report_date") == pl.col("report_date").max().over("utility_id_eia"))
47-
.filter(pl.col("entity_type").is_in(ENTITY_TYPES_EXCLUDE).not_())
4846
.group_by("utility_id_eia")
4947
.agg(
5048
pl.col("utility_name_eia").last().alias("utility_name"),

0 commit comments

Comments
 (0)