vizflow 0.5.8__py3-none-any.whl → 0.5.9__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. vizflow/ops.py +3 -2
  2. vizflow/schema_evolution.py +2 -0
  3. {vizflow-0.5.8.dist-info → vizflow-0.5.9.dist-info}/METADATA +1 -1
  4. {vizflow-0.5.8.dist-info → vizflow-0.5.9.dist-info}/RECORD +5 -5
  5. {vizflow-0.5.8.dist-info → vizflow-0.5.9.dist-info}/WHEEL +0 -0
vizflow/ops.py CHANGED
@@ -216,11 +216,12 @@ def forward_return(
216
216
  )
217
217
 
218
218
  # Add forward price and calculate return (guard against zero price)
219
+ # Explicitly cast to Float64 to ensure consistent schema across dates
219
220
  trade = joined.with_columns([
220
- pl.col("_forward_price").alias(forward_col),
221
+ pl.col("_forward_price").cast(pl.Float64).alias(forward_col),
221
222
  pl.when(pl.col(price_col) != 0)
222
223
  .then((pl.col("_forward_price") - pl.col(price_col)) / pl.col(price_col))
223
- .otherwise(pl.lit(None))
224
+ .otherwise(pl.lit(None, dtype=pl.Float64))
224
225
  .alias(return_col),
225
226
  ]).drop(["_forward_time", "_alpha_time", "_forward_price"])
226
227
 
vizflow/schema_evolution.py CHANGED
@@ -337,6 +337,8 @@ JYAO_V20251114 = SchemaEvolution(
337
337
  # Symbol column - parse_dtype for CSV, cast_dtype for feather/IPC
338
338
  # (feather files have embedded types, so cast is needed post-load)
339
339
  "ukey": ColumnSpec(parse_dtype=pl.Int64, cast_dtype=pl.Int64),
340
+ # Time columns
341
+ "ticktime": ColumnSpec(parse_dtype=pl.Int64, cast_dtype=pl.Int64),
340
342
  # Quote columns
341
343
  "BidPrice1": ColumnSpec(rename_to="bid_px0", parse_dtype=pl.Float64),
342
344
  "AskPrice1": ColumnSpec(rename_to="ask_px0", parse_dtype=pl.Float64),
{vizflow-0.5.8.dist-info → vizflow-0.5.9.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: vizflow
3
- Version: 0.5.8
3
+ Version: 0.5.9
4
4
  Requires-Python: >=3.9
5
5
  Requires-Dist: polars>=0.20.0
6
6
  Provides-Extra: dev
{vizflow-0.5.8.dist-info → vizflow-0.5.9.dist-info}/RECORD RENAMED
@@ -2,10 +2,10 @@ vizflow/__init__.py,sha256=KGksMU9wGsdGSbd2OgSHM2ofMZbTwgaCN92ZDEgASwU,614
2
2
  vizflow/config.py,sha256=nPZPXlqQbaY8u_FAdtPShvb0mdx3e2TRaQ2CILzliAU,7192
3
3
  vizflow/io.py,sha256=1T7t-L1ijrfEkE-gr4f45yiupJKA4-DxbJhsyN_tLnI,11939
4
4
  vizflow/market.py,sha256=MtNz_nnZxC66Aq-i2PXEwaFCTknijFWYZUUv6798k2s,2493
5
- vizflow/ops.py,sha256=QO8airzHFfKw8nGsStsecic2Z1dmq1rciSyAfSElh7M,10421
5
+ vizflow/ops.py,sha256=cUFeDdWfMk2hG8hpevTz_Dfnzk3fnhTI79UwMGzUkHw,10534
6
6
  vizflow/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
- vizflow/schema_evolution.py,sha256=3_qFIQJAgXUK0vQzanb355YVjytmfOACkelZlykIO8w,16349
7
+ vizflow/schema_evolution.py,sha256=HuyfHNGXUS82_oat1aSixd3msvOnBC9l5u0ec01blck,16447
8
8
  vizflow/viz.py,sha256=dzcY72hWMVbxWIyjwfUW3Ot3CunaP7O4GLVUzzOjkbY,1212
9
- vizflow-0.5.8.dist-info/METADATA,sha256=uapEZDP5ntZ1_a8uzexB_K45cIpK9rdOiqq87fSv_4M,388
10
- vizflow-0.5.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
11
- vizflow-0.5.8.dist-info/RECORD,,
9
+ vizflow-0.5.9.dist-info/METADATA,sha256=wrZtGCshKtCZDr96T8QfmYvdhBi7x2y0Bc_Fe5eAiC4,388
10
+ vizflow-0.5.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
11
+ vizflow-0.5.9.dist-info/RECORD,,