From 576fd3edee0e0c7f24de7875ee1ce6e3e1893cb9 Mon Sep 17 00:00:00 2001 From: Jayson Francis Date: Fri, 3 Jan 2025 20:12:48 -0800 Subject: [PATCH] final cleanup --- torchtitan/config_manager.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/torchtitan/config_manager.py b/torchtitan/config_manager.py index 97f09dd3..d7e1e7ee 100644 --- a/torchtitan/config_manager.py +++ b/torchtitan/config_manager.py @@ -7,7 +7,7 @@ import sys from dataclasses import asdict, dataclass, field, fields, is_dataclass -from typing import Any, Callable, Dict, List, Optional, Union +from typing import Any, Dict, List, Optional, Union import torch import tyro @@ -452,8 +452,7 @@ def update_instance(self, instance) -> None: def parse_args(self, args=None) -> None: """ - Parse CLI arguments, optionally load from a TOML file, - merge with defaults, and return a JobConfig instance. + Parse command line arguments and return the parsed args and the command line only args """ toml_data = self.maybe_load_toml(args or sys.argv[1:]) defaults = self.__class__