Skip to content

Commit 73ce0c0

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 6b3d8fe commit 73ce0c0

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

monai/networks/blocks/text_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def __init__(
5555
self.spatial_dims = spatial_dims
5656
if spatial_dims not in (2, 3):
5757
raise ValueError("spatial dimension should be 2 or 3.")
58-
58+
5959
if self.encoding == 'rand_embedding':
6060
self.text_embedding = nn.Embedding(out_channels, hidden_size)
6161
else:

tests/test_text_encoding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def test_test_encoding_shape(self):
3333
print(text_encoding.shape)
3434
self.assertEqual(text_encoding.shape, (32,256,1,1,1))
3535

36-
# test random enbedding
36+
# test random enbedding
3737
text_encoder = TextEncoder(spatial_dims=3, out_channels=32, encoding="rand_embedding", pretrained=True).to(device)
3838
text_encoding = text_encoder()
3939
print(text_encoding.shape)

0 commit comments

Comments
 (0)