refactor(zonebud): reapply OrderedDict to dict changes from PR modflo…

…wpy#1201
jdhughes-usgs · Aug 21, 2021 · 7cb08b0 · 7cb08b0
1 parent a0729c5
commit 7cb08b0
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/flopy/utils/zonbud.py b/flopy/utils/zonbud.py
@@ -151,9 +151,7 @@ def __init__(
 
         self.izone = izone
         self.allzones = np.unique(izone)
-        self._zonenamedict = dict(
-            [(z, "ZONE_{}".format(z)) for z in self.allzones]
-        )
+        self._zonenamedict = {z: "ZONE_{}".format(z) for z in self.allzones}
 
         if aliases is not None:
             s = (
@@ -2198,10 +2196,8 @@ def __init__(self, model, izone, extension=".zon", aliases=None):
         self._parent.add_package("zon", self)
         self.filename = self._parent.name + extension
         self.aliases = aliases
-        self.allzones = [int(zn) for zn in np.unique(izone) if zn != 0]
-        self._zonenamedict = dict(
-            [(zn, "ZONE_{}".format(zn)) for zn in self.allzones]
-        )
+        self.allzones = [int(z) for z in np.unique(izone) if z != 0]
+        self._zonenamedict = {z: "ZONE_{}".format(z) for z in self.allzones}
 
         if aliases is not None:
             if not isinstance(aliases, dict):