fix set context

oelarnes · oelarnes · commit 95e6783fc877 · 2025-03-08T22:47:22.000-05:00
diff --git a/spells/external.py b/spells/external.py
@@ -345,11 +345,11 @@ def get_set_context(set_code: str, force_download=False) -> int:
 
     df = summon(
         set_code,
-        columns=[ColName.NUM_DRAFTS],
+        columns=[ColName.NUM_TAKEN],
         group_by=[ColName.DRAFT_DATE, ColName.PICK_NUM],
     )
 
-    context_df = df.filter(pl.col(ColName.NUM_DRAFTS) > 1000).select(
+    context_df = df.filter(pl.col(ColName.NUM_TAKEN) > 1000).select(
         [
             pl.col(ColName.DRAFT_DATE).min().alias("release_date"),
             pl.col(ColName.PICK_NUM).max().alias("picks_per_pack"),

Original file line number	Diff line number	Diff line change
`@@ -345,11 +345,11 @@ def get_set_context(set_code: str, force_download=False) -> int:`
`345`	`345`
`346`	`346`	`df = summon(`
`347`	`347`	`set_code,`
`348`		`- columns=[ColName.NUM_DRAFTS],`
	`348`	`+ columns=[ColName.NUM_TAKEN],`
`349`	`349`	`group_by=[ColName.DRAFT_DATE, ColName.PICK_NUM],`
`350`	`350`	`)`
`351`	`351`
`352`		`- context_df = df.filter(pl.col(ColName.NUM_DRAFTS) > 1000).select(`
	`352`	`+ context_df = df.filter(pl.col(ColName.NUM_TAKEN) > 1000).select(`
`353`	`353`	`[`
`354`	`354`	`pl.col(ColName.DRAFT_DATE).min().alias("release_date"),`
`355`	`355`	`pl.col(ColName.PICK_NUM).max().alias("picks_per_pack"),`