summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xsetup.py2
-rw-r--r--szilagyi/dataset/SWI_-1.csv (renamed from dataset/SWI_-1.csv)0
-rw-r--r--szilagyi/dataset/SWI_-10.csv (renamed from dataset/SWI_-10.csv)0
-rw-r--r--szilagyi/dataset/SWI_-2.csv (renamed from dataset/SWI_-2.csv)0
-rw-r--r--szilagyi/dataset/SWI_-5.csv (renamed from dataset/SWI_-5.csv)0
-rw-r--r--szilagyi/dataset/SWI_0.csv (renamed from dataset/SWI_0.csv)0
-rw-r--r--szilagyi/dataset/SWI_1.csv (renamed from dataset/SWI_1.csv)0
-rw-r--r--szilagyi/dataset/SWI_10.csv (renamed from dataset/SWI_10.csv)0
-rw-r--r--szilagyi/dataset/SWI_2.csv (renamed from dataset/SWI_2.csv)0
-rw-r--r--szilagyi/dataset/SWI_5.csv (renamed from dataset/SWI_5.csv)0
-rw-r--r--szilagyi/plots.py4
11 files changed, 5 insertions, 1 deletions
diff --git a/setup.py b/setup.py
index 0a92236..6f89ba4 100755
--- a/setup.py
+++ b/setup.py
@@ -6,4 +6,6 @@ setup(
name="szilagyi",
packages=["szilagyi"],
extra_requires={"plots": ["matplotlib"]},
+ package_data={"": ["dataset/*.csv"]},
+ include_package_data=True,
)
diff --git a/dataset/SWI_-1.csv b/szilagyi/dataset/SWI_-1.csv
index 5c54050..5c54050 100644
--- a/dataset/SWI_-1.csv
+++ b/szilagyi/dataset/SWI_-1.csv
diff --git a/dataset/SWI_-10.csv b/szilagyi/dataset/SWI_-10.csv
index a97e2b1..a97e2b1 100644
--- a/dataset/SWI_-10.csv
+++ b/szilagyi/dataset/SWI_-10.csv
diff --git a/dataset/SWI_-2.csv b/szilagyi/dataset/SWI_-2.csv
index b8a35ff..b8a35ff 100644
--- a/dataset/SWI_-2.csv
+++ b/szilagyi/dataset/SWI_-2.csv
diff --git a/dataset/SWI_-5.csv b/szilagyi/dataset/SWI_-5.csv
index af7179e..af7179e 100644
--- a/dataset/SWI_-5.csv
+++ b/szilagyi/dataset/SWI_-5.csv
diff --git a/dataset/SWI_0.csv b/szilagyi/dataset/SWI_0.csv
index 3d19b98..3d19b98 100644
--- a/dataset/SWI_0.csv
+++ b/szilagyi/dataset/SWI_0.csv
diff --git a/dataset/SWI_1.csv b/szilagyi/dataset/SWI_1.csv
index a468a93..a468a93 100644
--- a/dataset/SWI_1.csv
+++ b/szilagyi/dataset/SWI_1.csv
diff --git a/dataset/SWI_10.csv b/szilagyi/dataset/SWI_10.csv
index 094f991..094f991 100644
--- a/dataset/SWI_10.csv
+++ b/szilagyi/dataset/SWI_10.csv
diff --git a/dataset/SWI_2.csv b/szilagyi/dataset/SWI_2.csv
index a1b042b..a1b042b 100644
--- a/dataset/SWI_2.csv
+++ b/szilagyi/dataset/SWI_2.csv
diff --git a/dataset/SWI_5.csv b/szilagyi/dataset/SWI_5.csv
index d87d202..d87d202 100644
--- a/dataset/SWI_5.csv
+++ b/szilagyi/dataset/SWI_5.csv
diff --git a/szilagyi/plots.py b/szilagyi/plots.py
index e0727c8..ed9b52c 100644
--- a/szilagyi/plots.py
+++ b/szilagyi/plots.py
@@ -1,3 +1,5 @@
+import os
+
import matplotlib.pyplot as plot
from .nomogram import *
@@ -12,7 +14,7 @@ def grid(start, end, steps):
if __name__ == "__main__":
- swis = load("dataset")
+ swis = load(os.path.join(os.path.dirname(__file__), "dataset"))
def _swi(x, y): # dt, depth
segments = find_boundary_curves(swis, x, y)