test

2025-06-13 14:45:59 -04:00 · 2025-06-13 14:45:59 -04:00 · e0fd6f00b4
commit e0fd6f00b4
parent 039afbb330
2 changed files with 4 additions and 4 deletions
--- 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
--- 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ˍ)