Add back autoplanner tests

Summary: As titled Differential Revision: D57866623
pytorch · May 28, 2024 · bbc3a69 · bbc3a69
1 parent 9a50de2
commit bbc3a69
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrec/distributed/planner/planners.py b/torchrec/distributed/planner/planners.py
@@ -760,7 +760,7 @@ def plan(
                 if not lowest_storage.fits_in(global_storage_constraints):
                     raise PlannerError(
                         error_type=PlannerErrorType.INSUFFICIENT_STORAGE,
-                        message="Unable to find a plan for this model in device_group {group}  because of insufficient storage. \n"
+                        message=f"Unable to find a plan for this model in device_group {group}  because of insufficient storage. \n"
                         + no_plan_solution
                         + last_planner_error_info,
                     )