From 5e307bd1be48665ea7fc7c8d0efefb7ca91c8bb9 Mon Sep 17 00:00:00 2001
From: HB_alon <hb-in@qq.com>
Date: Wed, 5 Dec 2018 22:16:44 -0800
Subject: [PATCH] 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
---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 2065462..ddc8b7e 100644
--- 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=[]),
     )
-- 
2.7.4