code: def import_module(module):  # pylint: disable=R0201
          
          if isinstance(module, list):
              all_modules = module
          else:
              all_modules = [module]
          for mod in all_modules:
              globals()[mod] = __import__(mod.strip())
docs: Import module that are needed for the code expr to compile.
  
          Argument:
            module (str or list): module(s) to import.