Fix dir(torch) for python 3.7 (#10271)
Summary:
fixes #10160.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10271
Differential Revision: D9188031
Pulled By: li-roy
fbshipit-source-id: a3620553a8ba2b7391acdf78dbe58afcdb6c5f7f
diff --git a/test/test_torch.py b/test/test_torch.py
index 8bcd30a..e494981 100644
--- a/test/test_torch.py
+++ b/test/test_torch.py
@@ -143,6 +143,9 @@
return tensors
+ def test_dir(self):
+ dir(torch)
+
def test_dot(self):
types = {
'torch.DoubleTensor': 1e-8,
diff --git a/torch/__init__.py b/torch/__init__.py
index 043ca11..e494cde 100644
--- a/torch/__init__.py
+++ b/torch/__init__.py
@@ -251,6 +251,8 @@
del manager_path
for name in dir(_C._VariableFunctions):
+ if name in ["__dir__", "__doc__"]:
+ continue
globals()[name] = getattr(_C._VariableFunctions, name)
################################################################################
