Avoid leaking has_torch_function and handle_torch_function in torch namespace (pytorch#46680) (pytorch#46719)

Summary: Pull Request resolved: pytorch#46680

Reviewed By: zou3519

Differential Revision: D24459823

Pulled By: albanD

fbshipit-source-id: 4ff6925afcf14214dc45921bca0d2f33ca1944a1

Author: albanD

torch/__init__.py

@@ -614,11 +614,12 @@ def compiled_with_cxx11_abi():
 quantized_lstm = torch.ops.aten.quantized_lstm
 quantized_gru = torch.ops.aten.quantized_gru
 
-from .overrides import has_torch_function, handle_torch_function
 
 def Assert(condition, message):
     r"""A wrapper around Python's assert which is symbolically traceable.
     """
+    from .overrides import has_torch_function, handle_torch_function
+
     if type(condition) is not torch.Tensor and has_torch_function((condition,)):
         return handle_torch_function(Assert, (condition,), condition, message)
     assert condition, message