spells-mtg 0.11.4__py3-none-any.whl → 0.11.6__py3-none-any.whl

spells/card_data_files.py

@@ -28,7 +28,6 @@ ratings_col_defs = {
     ColName.NAME: pl.col("name").cast(pl.String),
     ColName.COLOR: pl.col("color").cast(pl.String),
     ColName.RARITY: pl.col("rarity").cast(pl.String),
-    ColName.CARD_TYPE: pl.col("types").cast(pl.String),
     ColName.IMAGE_URL: pl.col("url").cast(pl.String),
     ColName.NUM_SEEN: pl.col("seen_count").cast(pl.Int64),
     ColName.LAST_SEEN: pl.col("seen_count") * pl.col("avg_seen").cast(pl.Float64),
@@ -41,7 +40,8 @@ ratings_col_defs = {
     ColName.WON_OPENING_HAND: pl.col("opening_hand_game_count")
     * pl.col("opening_hand_win_rate").cast(pl.Float64),
     ColName.DRAWN: pl.col("drawn_game_count").cast(pl.Int64),
-    ColName.WON_DRAWN: pl.col("drawn_win_rate") * pl.col("drawn_game_count").cast(pl.Float64),
+    ColName.WON_DRAWN: pl.col("drawn_win_rate")
+    * pl.col("drawn_game_count").cast(pl.Float64),
     ColName.NUM_GIH: pl.col("ever_drawn_game_count").cast(pl.Int64),
     ColName.NUM_GIH_WON: pl.col("ever_drawn_game_count")
     * pl.col("ever_drawn_win_rate").cast(pl.Float64),
@@ -107,9 +107,9 @@ def deck_color_df(
             [
                 pl.lit(set_code).alias(ColName.EXPANSION),
                 pl.lit(format).alias(ColName.EVENT_TYPE),
-                (pl.lit("Top") if player_cohort == "top" else pl.lit(None)).alias(
-                    ColName.PLAYER_COHORT
-                ).cast(pl.String),
+                (pl.lit("Top") if player_cohort == "top" else pl.lit(None))
+                .alias(ColName.PLAYER_COHORT)
+                .cast(pl.String),
                 *[val.alias(key) for key, val in deck_color_col_defs.items()],
             ]
         )
@@ -172,21 +172,24 @@ def base_ratings_df(
                 out=ratings_file_path,
             )
 
-        concat_list.append(pl.read_json(ratings_file_path).with_columns(
-            (pl.lit(deck_color) if deck_color != "any" else pl.lit(None)).alias(
-                ColName.MAIN_COLORS
-            ).cast(pl.String)
-        ))
-    df = pl.concat(concat_list)
-
-    return df.select(
-        [
-            pl.lit(set_code).alias(ColName.EXPANSION),
-            pl.lit(format).alias(ColName.EVENT_TYPE),
-            (pl.lit("Top") if player_cohort == "top" else pl.lit(None)).alias(
-                ColName.PLAYER_COHORT
-            ).cast(pl.String),
-            ColName.MAIN_COLORS,
-            *[val.alias(key) for key, val in ratings_col_defs.items()],
-        ]
-    )
+        concat_list.append(
+            pl.read_json(ratings_file_path, infer_schema_length=500)
+            .with_columns(
+                (pl.lit(deck_color) if deck_color != "any" else pl.lit(None))
+                .alias(ColName.MAIN_COLORS)
+                .cast(pl.String)
+            )
+            .select(
+                [
+                    pl.lit(set_code).alias(ColName.EXPANSION),
+                    pl.lit(format).alias(ColName.EVENT_TYPE),
+                    (pl.lit("Top") if player_cohort == "top" else pl.lit(None))
+                    .alias(ColName.PLAYER_COHORT)
+                    .cast(pl.String),
+                    ColName.MAIN_COLORS,
+                    *[val.alias(key) for key, val in ratings_col_defs.items()],
+                ]
+            )
+        )
+
+    return pl.concat(concat_list)

{spells_mtg-0.11.4.dist-info → spells_mtg-0.11.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spells-mtg
-Version: 0.11.4
+Version: 0.11.6
 Summary: analaysis of 17Lands.com public datasets
 Author-Email: Joel Barnes <oelarnes@gmail.com>
 License: MIT

{spells_mtg-0.11.4.dist-info → spells_mtg-0.11.6.dist-info}/RECORD

@@ -3,7 +3,7 @@ spells/.ruff_cache/0.8.6/17785301476771359756,sha256=gPYLG8psuOSo37IefLzTu5wgRrb
 spells/.ruff_cache/CACHEDIR.TAG,sha256=WVMVbX4MVkpCclExbq8m-IcOZIOuIZf5FrYw5Pk-Ma4,43
 spells/__init__.py,sha256=0pnh2NLn9FrNKncE9-tBsighp3e8YAsN--_ovUpgWGs,333
 spells/cache.py,sha256=6cl0q62erR3LCANPSfxG5-J7JQfLwNdWjzlBpfiL4IE,7174
-spells/card_data_files.py,sha256=yaZzXKJgrsg__Kov7eKLosx4oyexMClV3a7Tfxdlk3w,6433
+spells/card_data_files.py,sha256=k9NcYLgK5vdqLfM_VZ0XMnNGWBO-IloocDXg5J8bO-s,6526
 spells/cards.py,sha256=6stFPhJOzHqvQnkSv9cDeylRa_7L9Y8sOaowiZhzz6I,4174
 spells/columns.py,sha256=s_PYyg2QaRL6kLWFNKCBEfbMF0x7O7-Yd9SeG1ttYL4,18206
 spells/config.py,sha256=zpRUZ-6JKALE149L0yfeD1AgqKPaVBGHfD_TdHi-jqE,235
@@ -16,8 +16,8 @@ spells/log.py,sha256=3avmg65hru8K9npKLvPp1wWWxq-hoEYDUCbxqhPkKUw,2175
 spells/manifest.py,sha256=ExWVk17BRw615UmvrV817xwz457yfTNdNMNE_M00aEg,8338
 spells/schema.py,sha256=DbMvV8PIThJTp0Xzp_XIorlW6JhE1ud1kWRGf5SQ4_c,6406
 spells/utils.py,sha256=IO3brrXVvZla0LRTEB5v6NgGqZb_rYA46XtKBURGMNk,1944
-spells_mtg-0.11.4.dist-info/METADATA,sha256=4U7yb5Kz5Pku2AD8BjZe9WKbvneYwK0NtqRI7ydA8GQ,47369
-spells_mtg-0.11.4.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-spells_mtg-0.11.4.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
-spells_mtg-0.11.4.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
-spells_mtg-0.11.4.dist-info/RECORD,,
+spells_mtg-0.11.6.dist-info/METADATA,sha256=X6UVl1fVGPdadQffDujjgizXMPINGPixenDm3mTfVKU,47369
+spells_mtg-0.11.6.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+spells_mtg-0.11.6.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
+spells_mtg-0.11.6.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
+spells_mtg-0.11.6.dist-info/RECORD,,