Add random seed to collect-data.py

Author: petebachant

scripts/collect-data.py

@@ -5,13 +5,14 @@
 import numpy as np
 import pandas as pd
 
-if __name__ == "__main__":
-    df = pd.DataFrame(
-        data={"voltage": np.random.randn(1000)},
-        index=pd.date_range(
-            "2024-12-01", "2024-12-01 00:00:01", freq="ms", inclusive="left"
-        ),
-    )
-    df.index.name = "time"
-    os.makedirs("data/raw", exist_ok=True)
-    df.to_csv("data/raw/data.csv", index=True)
+np.random.seed(4545)
+
+df = pd.DataFrame(
+    data={"voltage": np.random.randn(1000)},
+    index=pd.date_range(
+        "2024-12-01", "2024-12-01 00:00:01", freq="ms", inclusive="left"
+    ),
+)
+df.index.name = "time"
+os.makedirs("data/raw", exist_ok=True)
+df.to_csv("data/raw/data.csv", index=True)