Skip to content

Commit d43109d

Browse files
committed
Fix typo
1 parent 5fd265b commit d43109d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_base.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66

77

88
def test_from_pretrained_with_mismatched_keys():
9-
orginal_model = smp.Unet(classes=1)
9+
original_model = smp.Unet(classes=1)
1010

1111
with tempfile.TemporaryDirectory() as temp_dir:
12-
orginal_model.save_pretrained(temp_dir)
12+
original_model.save_pretrained(temp_dir)
1313

1414
# we should catch warning here and check if there specific keys there
1515
with pytest.warns(UserWarning):
@@ -18,7 +18,7 @@ def test_from_pretrained_with_mismatched_keys():
1818
assert restored_model.segmentation_head[0].out_channels == 2
1919

2020
# verify all the weight are the same expect mismatched ones
21-
original_state_dict = orginal_model.state_dict()
21+
original_state_dict = original_model.state_dict()
2222
restored_state_dict = restored_model.state_dict()
2323

2424
expected_mismatched_keys = [

0 commit comments

Comments
 (0)