[webdata] stats df

oist · Jul 2, 2024 · 1b727fa · 1b727fa
1 parent 823c1d8
commit 1b727fa
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/scripts/make_web_data.py b/scripts/make_web_data.py
@@ -107,11 +107,14 @@ def main(
     write_dir: Optional[Path] = None,
     length: int = 100,
 ) -> None:
-    stats_df, ldf = _make_stats_df(profile_and_rewards_path)
-    log_path = profile_and_rewards_path.parent.expanduser()
     if write_dir is None:
         write_dir = Path("saved-web-data")
 
+    stats_df, ldf = _make_stats_df(profile_and_rewards_path)
+    stats_df.write_parquet(write_dir / "stats.parqut", compression="snappy")
+
+    log_path = profile_and_rewards_path.parent.expanduser()
+
     for point in starting_points:
         index = point // 1024000
         ldfi = ldf.filter(
@@ -125,10 +128,12 @@ def main(
             index * 1024000,
         )
         cxy_df.write_parquet(
-            write_dir / f"saved_cpos-{point}.parqut", compression="snappy"
+            write_dir / f"saved_cpos-{point}.parqut",
+            compression="snappy",
         )
         sxy_df.write_parquet(
-            write_dir / f"saved_spos-{point}.parqut", compression="snappy"
+            write_dir / f"saved_spos-{point}.parqut",
+            compression="snappy",
         )