move sample entry point to app

microsoft · Dec 26, 2024 · fa21c04 · fa21c04
1 parent d3e3f60
commit fa21c04
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/rdagent/app/data_science/debug.py b/rdagent/app/data_science/debug.py
@@ -1,4 +1,7 @@
-"""
-entry of debuggging function
-- e.g. creating debug datasets
-"""
+import fire
+
+from rdagent.scenarios.data_science.debug.data import create_debug_data
+
+
+if __name__ == "__main__":
+    fire.Fire(create_debug_data)
diff --git a/rdagent/scenarios/data_science/debug/data.py b/rdagent/scenarios/data_science/debug/data.py
@@ -2,7 +2,6 @@
 from pathlib import Path
 import platform
 import pandas as pd
-import fire
 import shutil
 
 from rdagent.app.kaggle.conf import KAGGLE_IMPLEMENT_SETTING
@@ -166,5 +165,3 @@ def create_debug_data(
             print(f"Error processing {file_path}: {e}")
             continue
 
-if __name__ == "__main__":
-    fire.Fire(create_debug_data)