diff options
Diffstat (limited to 'szilagyi/plots.py')
-rw-r--r-- | szilagyi/plots.py | 4 |
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) |