work on pip package

acestep/data_sampler.py

@@ -0,0 +1,21 @@
+import json
+from pathlib import Path
+import random
+
+
+DEFAULT_ROOT_DIR = "examples/input_params"
+
+
+class DataSampler:
+    def __init__(self, root_dir=DEFAULT_ROOT_DIR):
+        self.root_dir = root_dir
+        self.input_params_files = list(Path(self.root_dir).glob("*.json"))
+
+    def load_json(self, file_path):
+        with open(file_path, "r", encoding="utf-8") as f:
+            return json.load(f)
+
+    def sample(self):
+        json_path = random.choice(self.input_params_files)
+        json_data = self.load_json(json_path)
+        return json_data