summaryrefslogtreecommitdiff
path: root/szilagyi/plots.py
diff options
context:
space:
mode:
authorAki <please@ignore.pl>2022-08-30 00:39:01 +0200
committerAki <please@ignore.pl>2022-08-30 00:39:01 +0200
commit9f125795484a73593a2631cb4256475864b0badf (patch)
tree8f37da0003cd96bac2f102d0550dfb1f94681d6c /szilagyi/plots.py
parente26167b82ea6b6c8ec049321a7426bfe076709ac (diff)
downloadszilagyi-9f125795484a73593a2631cb4256475864b0badf.zip
szilagyi-9f125795484a73593a2631cb4256475864b0badf.tar.gz
szilagyi-9f125795484a73593a2631cb4256475864b0badf.tar.bz2
Included dataset in the package
Diffstat (limited to 'szilagyi/plots.py')
-rw-r--r--szilagyi/plots.py4
1 files changed, 3 insertions, 1 deletions
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)