From e0fd6f00b412557923233a5766f31618e5df07fb Mon Sep 17 00:00:00 2001
From: Liyu Xiao <liyu@qoherent.ai>
Date: Fri, 13 Jun 2025 14:45:59 -0400
Subject: [PATCH] test

---
 requirements.txt                     | 7 +++----
 scripts/dataset_building/data_gen.py | 1 +
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 4a02db0..0eaaff2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,13 +1,12 @@
 h5py
-lightning
+pytorch-lightning
 matplotlib
 numpy
 PyYAML
-PyYAML
-scikit_learn
+scikit-learn
 timm
 torch
 onnx
 onnxruntime
---index-url https://riahub.qoherent.internal:3000/api/packages/qoherent/pypi/simple/
+--index-url https://git.riahub.ai/api/packages/qoherent/pypi/simple/ 
 utils
diff --git a/scripts/dataset_building/data_gen.py b/scripts/dataset_building/data_gen.py
index 27d09b7..7380f97 100644
--- a/scripts/dataset_building/data_gen.py
+++ b/scripts/dataset_building/data_gen.py
@@ -75,3 +75,4 @@ if __name__ == "__main__":
     )
     args = p.parse_args()
     generate_modulated_signals(args.output_dir)
+    print("generated data to " + args.output_dirˍ)