Skip to content

Commit 702e740

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 1667eb7 commit 702e740

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_bundle_ckpt_export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def tearDown(self):
6262

6363
@parameterized.expand([TEST_CASE_1, TEST_CASE_2, TEST_CASE_3])
6464
def test_export(self, key_in_ckpt, use_trace):
65-
save_state(src=self.net if key_in_ckpt == "" else {key_in_ckpt: self.net}, path=self.ckpt_file) # noqa: E117
65+
save_state(src=self.net if key_in_ckpt == "" else {key_in_ckpt: self.net}, path=self.ckpt_file)
6666
full_cmd = self.cmd + [self.ckpt_file, "--key_in_ckpt", key_in_ckpt, "--args_file", self.def_args_file]
6767
if use_trace == "True":
6868
full_cmd += ["--use_trace", use_trace, "--input_shape", "[1, 1, 96, 96, 96]"]

0 commit comments

Comments
 (0)