Skip to content

Commit 210e914

Browse files
committed
Fix #7 - remove non-utilities from selection
1 parent 8e4bd3a commit 210e914

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tariff_fetch/cli.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def prompt_utility(state: str) -> Utility:
4343
pl.read_parquet(CORE_EIA861_Yearly_Sales.https)
4444
.filter(pl.col("state") == state.upper())
4545
.filter(pl.col("report_date") == pl.col("report_date").max().over("utility_id_eia"))
46+
.filter(pl.col("entity_type").is_in(ENTITY_TYPES_SORTORDER))
4647
.group_by("utility_id_eia")
4748
.agg(
4849
pl.col("utility_name_eia").last().alias("utility_name"),

0 commit comments

Comments
 (0)