Possible fix for einops warning (pytorch#124084)

See arogozhnikov/einops#315

Pull Request resolved: pytorch#124084
Approved by: https://github.com/peterbell10

torch/_dynamo/trace_rules.py

@@ -3036,10 +3036,6 @@ def add_module_init_func(name: str, init_func: Callable[[], None]) -> None:
  """Register a module without eagerly importing it"""
  # If the module is already imported, eagerly run init
  assert "." not in name, f"Expected a root module name, but got {name}"
- if name in sys.modules:
- init_func()
-
- # Module is not yet imported, delay processing until needed
  assert name not in _lazy_module_init
  _lazy_module_init[name].append(init_func)