Skip to content

Commit

Permalink
馃 Lint code
Browse files Browse the repository at this point in the history
  • Loading branch information
ivy-branch committed May 21, 2023
1 parent e2d4f16 commit 07bbd25
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 9 deletions.
9 changes: 2 additions & 7 deletions ivy_builder/abstract/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -469,7 +469,7 @@ def _save_spec_to_disk(self):
save_dict_as_json(spec_args_dict, complete_spec_args_filepath)

def _save_info_to_disk(self):
info_dir = os.path.join(self._spec.log_dir, "info")
info_dir = os.path.join(self._spec.log_dir, "info")
os.makedirs(info_dir, exist_ok=True)
info_filepath = _get_valid_filepath(info_dir, "info", ".txt")
if not ivy.exists(git):
Expand All @@ -479,12 +479,7 @@ def _save_info_to_disk(self):
)
return
with open(info_filepath, "w+") as info_file:
info_file.writelines(
[
"time of execution:\n",
str(datetime.now()) + "\n\n"
]
)
info_file.writelines(["time of execution:\n", str(datetime.now()) + "\n\n"])

def _initialize_model(self, checkpoint_path=None):
self._pre_init()
Expand Down
1 change: 0 additions & 1 deletion ivy_builder_tests/test_trainers.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,6 @@ def test_reduced_cost_after_checkpoint_load(dev_str, compile_mode, fw):
os.path.join(os.path.dirname(os.path.abspath(__file__)), "../ivy_builder_demos")
)


compile_mode = compile_mode

# dataset dirs specification
Expand Down
2 changes: 1 addition & 1 deletion ivy_builder_tests/test_tune.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@

def test_tune_numeric_spec(dev_str, fw):
if fw == "numpy":
pytest.skip()
pytest.skip()
builder_helpers.remove_dirs()
data_loader_spec_args = {"batch_size": 1}
trainer_spec_args = {
Expand Down

0 comments on commit 07bbd25

Please sign in to comment.