Skip to content

Commit

Permalink
Fix feedback from PR 111 (#113)
Browse files Browse the repository at this point in the history
Co-authored-by: gnadathur <[email protected]>
  • Loading branch information
gnadathur and gnadathur authored Mar 5, 2024
1 parent c5a4308 commit 5eeb185
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 1 deletion.
8 changes: 8 additions & 0 deletions test/test_job_config.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.

import tempfile

import pytest
from torchtrain.config_manager import JobConfig

Expand All @@ -20,3 +22,9 @@ def test_job_file_does_not_exist(self):
with pytest.raises(FileNotFoundError):
config = JobConfig()
config.parse_args(["--job.config_file", "ohno.toml"])

def test_empty_config_file(self):
with tempfile.NamedTemporaryFile() as fp:
config = JobConfig()
config.parse_args(["--job.config_file", fp.name])
assert config.job.description
1 change: 0 additions & 1 deletion torchtrain/config_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -227,7 +227,6 @@ def init_args_from_command_line(
)
parser.add_argument(
"--training.enable_selective_ac",
default=False,
action="store_true",
help="whether to enable selective activation checkpointing",
)
Expand Down

0 comments on commit 5eeb185

Please sign in to comment.