diff options
Diffstat (limited to 'szilagyi/plots.py')
-rw-r--r-- | szilagyi/plots.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/szilagyi/plots.py b/szilagyi/plots.py index 8b325d6..9e0c2c5 100644 --- a/szilagyi/plots.py +++ b/szilagyi/plots.py @@ -2,6 +2,7 @@ import os import matplotlib.pyplot as plot +from . import _dataset from .nomogram import * @@ -14,7 +15,7 @@ def grid(start, end, steps): if __name__ == "__main__": - swis = load(os.path.join(os.path.dirname(__file__), "_dataset")) + swis = _dataset.load() def _swi(x, y): # dt, depth segments = find_boundary_curves(swis, x, y) |