Merge pull request #80 from mbakker7/75-buildingpit-layers

fix #75
mbakker7 · Sep 10, 2023 · b2cb772 · b2cb772
2 parents e0ecd50 + 9d4edcd
commit b2cb772
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/timml/inhomogeneity.py b/timml/inhomogeneity.py
@@ -376,7 +376,7 @@ def __init__(
         self.order = order
         self.ndeg = ndeg
 
-        self.layers = layers  # layers with impermeable wall
+        self.layers = np.atleast_1d(layers)  # layers with impermeable wall
         self.nonimplayers = list(
             set(range(self.model.aq.naq)) - set(self.layers)
         )  # layers without wall

diff --git a/timml/trace.py b/timml/trace.py
@@ -24,7 +24,7 @@ def timtraceline(
     silent=False,
     returnlayers=False,
     *,
-    metadata=False
+    metadata=False,
 ):
     verbose = False  # used for debugging
     if not metadata:
@@ -257,7 +257,7 @@ def timtracelines(
     silent=".",
     win=[-1e30, 1e30, -1e30, 1e30],
     *,
-    metadata=False
+    metadata=False,
 ):
     xyztlist = []
     for x, y, z in zip(xstart, ystart, zstart):

diff --git a/timml/util.py b/timml/util.py
@@ -96,7 +96,7 @@ def contour(
         newfig=True,
         figsize=None,
         legend=True,
-        **kwargs
+        **kwargs,
     ):
         """Contour plot
 
@@ -245,7 +245,7 @@ def tracelines(
         figsize=None,
         *,
         return_traces=False,
-        metadata=False
+        metadata=False,
     ):
         """Draw trace lines"""
         if color is None:

diff --git a/timml/well.py b/timml/well.py
@@ -177,7 +177,7 @@ def capzone(
         nstepmax=100,
         silent=".",
         *,
-        metadata=False
+        metadata=False,
     ):
         """Compute a capture zone
 
@@ -249,7 +249,7 @@ def plotcapzone(
         figsize=None,
         *,
         return_traces=False,
-        metadata=False
+        metadata=False,
     ):
         """Plot a capture zone