File tree Expand file tree Collapse file tree 2 files changed +13
-1
lines changed
src/tuning_config_recommender Expand file tree Collapse file tree 2 files changed +13
-1
lines changed Original file line number Diff line number Diff line change 88** /* .egg-info
99** /values-internal.yaml
1010** /values- * .yaml
11- ** /values-cli-internal.yaml
11+ ** /values-api-internal *
12+ ** /values-cli-internal *
Original file line number Diff line number Diff line change @@ -83,6 +83,15 @@ def _populate_data_config(self, data_paths: list[str]):
8383 ],
8484 }
8585
86+ def _resolve_data_paths_in_data_config (self , data_config ):
87+ import glob
88+
89+ for dataset in data_config .get ("datasets" , []):
90+ dataset ["data_paths" ] = [
91+ p for path in dataset .get ("data_paths" , []) for p in glob .glob (path )
92+ ]
93+ return data_config
94+
8695 def execute (
8796 self ,
8897 tuning_config ,
@@ -103,6 +112,7 @@ def execute(
103112 if "_data" in path :
104113 data_paths .append (path )
105114 data_config = self ._populate_data_config (data_paths )
115+ data_config = self ._resolve_data_paths_in_data_config (data_config )
106116 ir , patches = super ().execute (
107117 tuning_config ,
108118 compute_config ,
@@ -111,6 +121,7 @@ def execute(
111121 unique_tag ,
112122 skip_estimator ,
113123 )
124+
114125 ir = ir .to_dict ()
115126 target_dir = (self .base_dir / unique_tag ).resolve ()
116127 target_dir .mkdir (parents = True , exist_ok = True )
You can’t perform that action at this time.
0 commit comments