[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

monai/networks/blocks/text_embedding.py

@@ -55,7 +55,7 @@ def __init__(
         self.spatial_dims = spatial_dims
         if spatial_dims not in (2, 3):
             raise ValueError("spatial dimension should be 2 or 3.")
-        
+
         if self.encoding == 'rand_embedding':
             self.text_embedding = nn.Embedding(out_channels, hidden_size)
         elif self.encoding == 'clip_embedding':
@@ -64,7 +64,7 @@ def __init__(
                 model_url = url_map["clip_encoding_univeral_model_31"]
                 pretrain_state_dict = model_zoo.load_url(model_url)
                 self.text_embedding.data = pretrain_state_dict.float()
-                print('load word embedding: {}'.format(self.encoding))
+                print(f'load word embedding: {self.encoding}')
             self.text_to_vision = nn.Linear(text_dim, hidden_size)
         else:
             raise Exception(f'{self.encoding} is not implemented, please add your own')

tests/test_text_encoding.py

@@ -33,7 +33,7 @@ def test_test_encoding_shape(self):
         print(text_encoding.shape)
         self.assertEqual(text_encoding.shape, (32,256,1,1,1))
 
-        # test random enbedding 
+        # test random enbedding
         text_encoder = TextEncoder(spatial_dims=3, out_channels=32, encoding="rand_embedding", pretrained=True).to(device)
         text_encoding = text_encoder()
         print(text_encoding.shape)