update setup.py

Author: carloalbertobarbano

setup.py

@@ -8,7 +8,7 @@
     author='Carlo Alberto Barbano',
     author_email='[email protected]',
     license='MIT',
-    packages=['torchstain'],
+    packages=find_packages(exclude=('tests')),
     zip_safe=False,
     install_requires=[
         'torch',