fix setup

pytorch/edgeml_pytorch/graph/rnn.py

@@ -324,8 +324,6 @@ class FastGRNNCUDACell(RNNCell):
             raise Exception('FastGRNNCUDACell is supported only on GPU devices.')
         self._input_size = input_size
         self._hidden_size = hidden_size
-        self._gate_non_linearity = gate_non_linearity
-        self._update_non_linearity = update_non_linearity
         self._zetaInit = zetaInit
         self._nuInit = nuInit
         self._name = name

pytorch/setup.py

@@ -4,7 +4,7 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 from edgeml_pytorch.utils import findCUDA
 
 if findCUDA() is not None:
-    setup(
+    setuptools.setup(
         name='fastgrnn_cuda',
         ext_modules=[
             CUDAExtension('fastgrnn_cuda', [