code: def test_init__no_dim_warning(self) -> None:
      with pytest.warns(
          UserWarning,
          match=re.escape(
              "Memory bank size 'size=10' does not specify feature "
              "dimension. It is recommended to set the feature dimension with "
              "'size=(n, dim)' when creating the memory bank. Distributed "
              "training might fail if the feature dimension is not set."
          ),
      ):
          MemoryBankModule(size=10)
  
file_path: lightly/tests/models/modules/test_memory_bank.py
imports: import re
  import unittest
  import pytest
  import torch
  from lightly.models.modules.memory_bank import MemoryBankModule
label_desc: Use image node_id 2 for calling the TestMemoryBank obj's underlying member method code with example usage: obj.test_init__no_dim_warning() without return types