1 files changed, 10 insertions, 6 deletions
diff --git a/setup.py b/setup.py
index f00e934..366f36c 100755
--- a/setup.py
+++ b/setup.py
@@ -1,10 +1,13 @@
 #!/usr/bin/env python
 
-from setuptools import setup
-from codecs import open
-from os import path
+import setuptools
 
-setup(
+INSTALL_REQUIRES = ["paramiko", "wheel", "wget",
+                    'repoze.lru; python_version<"3.2"']
+EXTRAS_REQUIRE = {"NumpyAcceleration": ["numpy"]}
+
+
+setuptools.setup(
     name="Rumba",
     version="0.4",
     url="https://gitlab.com/arcfire/rumba",
@@ -14,6 +17,7 @@ setup(
     license="LGPL",
     description="Rumba measurement framework for RINA",
     packages=["rumba", "rumba.testbeds", "rumba.prototypes"],
-    install_requires=["paramiko", "wheel", "wget"],
-    scripts = ['tools/rumba-access']
+    install_requires=INSTALL_REQUIRES,
+    extras_require=EXTRAS_REQUIRE,
+    scripts=['tools/rumba-access']
 )