Skip to content

Commit f40fbd3

Browse files
committed
fix: several fixes for API
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
1 parent ec8eb4f commit f40fbd3

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

.gitignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,5 @@
88
**/*.egg-info
99
**/values-internal.yaml
1010
**/values-*.yaml
11-
**/values-cli-internal.yaml
11+
**/values-api-internal*
12+
**/values-cli-internal*

src/tuning_config_recommender/adapters.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)