add remix and data sample
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
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
|
||||
|
||||
# glob
|
||||
self.input_params_files = list(Path(self.root_dir).glob("*.json"))
|
||||
|
||||
def load_json(self, file_path):
|
||||
with open(file_path, "r") 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
|
||||
Reference in New Issue
Block a user