Skip to content

Commit 61c6921

Browse files
fix: fms config adapter wrapper (#661)
Signed-off-by: yashasvi <yashasvi@ibm.com>
1 parent 747d10f commit 61c6921

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

tuning/fms-recommender.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,9 @@
88
import subprocess
99

1010
# Third Party
11-
from recommender.adapters import FMSAdapter # pylint: disable=import-error
11+
from tuning_config_recommender.adapters import ( # pylint: disable=import-error
12+
FMSAdapter,
13+
)
1214
import yaml
1315

1416
ACCEL_NESTED_PREFIXES = {
@@ -132,15 +134,15 @@ def main():
132134
return
133135

134136
adapter = FMSAdapter(base_dir=Path("fms_recommender_ouput/final"))
135-
ir, patches = adapter.execute(
137+
result = adapter.execute(
136138
train_config=train_cfg,
137139
dist_config=dist_cfg,
138140
compute_config={},
139141
data_config=data_cfg,
140142
unique_tag="fms-recommender",
143+
paths={},
141144
)
142-
out = adapter._to_target(ir, patches, tag="fms-recommender")
143-
launch_cmd = out["launch_command"]
145+
launch_cmd = result["launch_command"]
144146

145147
if args.preview:
146148
print("\n[LAUNCH COMMAND — PREVIEW ONLY]\n")

0 commit comments

Comments
 (0)