Fix Invalid syntax at convert_recipe_to_code.py (#1642)

* Fix Invalid syntax at convert_recipe_to_code.py * Fix Invalid syntax at convert_recipe_to_code.py --------- Co-authored-by: Eugene Khvedchenya <[email protected]>
Deci-AI · Nov 20, 2023 · 8e9f9d4 · 8e9f9d4
1 parent c5a7e05
commit 8e9f9d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/super_gradients/convert_recipe_to_code.py b/src/super_gradients/convert_recipe_to_code.py
@@ -179,13 +179,13 @@ def main():
     )
 
     train_dataloader = dataloaders.get(
-        name={train_dataloader},
+        name={wrap_in_quotes_if_string(train_dataloader)},
         dataset_params={train_dataset_params},
         dataloader_params={train_dataloader_params},
     )
 
     val_dataloader = dataloaders.get(
-        name={val_dataloader},
+        name={wrap_in_quotes_if_string(val_dataloader)},
         dataset_params={val_dataset_params},
         dataloader_params={val_dataloader_params},
     )