From c8dc1fe3c4357bceede84443bc4d7ddc95732e75 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=ABl=20Abrahams?= <JSAbrahams@users.noreply.github.com>
Date: Thu, 23 Jan 2020 22:50:47 +0100
Subject: [PATCH] Specify data file in setup

This means that the data file is included when setting up the package locally.
---
 setup.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/setup.py b/setup.py
index f8cf4a3..bb472eb 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup
+import os
 
 setup(
    name='prtools',
@@ -8,4 +9,7 @@
    author_email='',
    packages=['prtools'],
    install_requires=['sklearn', 'numpy', 'matplotlib', 'requests', 'mlxtend'],
+   package_data={
+      'prtools': [ os.path.join('data', '*.mat') ],
+   },
 )