use "Extension" instead of the unimported "setuptools.Extension" (#14475)

Summary:
use "Extension" instead of the unimported "setuptools.Extension"
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14475

Differential Revision: D13356219

Pulled By: ezyang

fbshipit-source-id: 5a3e7eb73a32d6bf09676efd9eddded5586435cd

diff --git a/setup.py b/setup.py
index 2065462..ddc8b7e 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -901,13 +901,13 @@ if USE_CUDA:
 # These extensions are built by cmake and copied manually in build_extensions()
 # inside the build_ext implementaiton
 extensions.append(
-    setuptools.Extension(
+    Extension(
         name=str('caffe2.python.caffe2_pybind11_state'),
         sources=[]),
 )
 if USE_CUDA:
     extensions.append(
-        setuptools.Extension(
+        Extension(
             name=str('caffe2.python.caffe2_pybind11_state_gpu'),
             sources=[]),
     )