resolve #229

pysal · Jul 15, 2024 · fb33b0e · fb33b0e
1 parent a67525a
commit fb33b0e
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 6 deletions.
diff --git a/giddy/directional.py b/giddy/directional.py
@@ -360,7 +360,7 @@ def plot_origin(self):  # TODO add attribute option to color vectors
 
         xlim = [self._dx.min(), self._dx.max()]
         ylim = [self._dy.min(), self._dy.max()]
-        for x, y in zip(self._dx, self._dy):
+        for x, y in zip(self._dx, self._dy, strict=False):
             xs = [0, x]
             ys = [0, y]
             plt.plot(xs, ys, "-b")  # TODO change this to scale with attribute

diff --git a/giddy/markov.py b/giddy/markov.py
@@ -817,7 +817,7 @@ def __init__(
             self.classes = classes
             self.k = len(classes)
             self.m = self.k
-            label_dict = dict(zip(classes, range(self.k)))
+            label_dict = dict(zip(classes, range(self.k), strict=False))
             y_int = []
             for yi in y:
                 y_int.append(list(map(label_dict.get, yi)))

diff --git a/giddy/sequence.py b/giddy/sequence.py
@@ -185,7 +185,7 @@ def __init__(self, y, subs_mat=None, dist_type=None, indel=None, cluster_type=No
         self.indel = indel
         self.subs_mat = subs_mat
         self.cluster_type = cluster_type
-        self.label_dict = dict(zip(self.classes, range(self.k)))
+        self.label_dict = dict(zip(self.classes, range(self.k), strict=False))
 
         y_int = []
         for yi in y:
@@ -267,7 +267,11 @@ def __init__(self, y, subs_mat=None, dist_type=None, indel=None, cluster_type=No
                     y_tran_index = np.zeros_like(y_int)
                     y_tran = []
                     for i in range(y_int.shape[1]):
-                        y_tran.append(list(zip(y_int_ext[:, i], y_int_ext[:, i + 1])))
+                        y_tran.append(
+                            list(
+                                zip(y_int_ext[:, i], y_int_ext[:, i + 1], strict=False)
+                            )
+                        )
                     for i in range(y_int.shape[0]):
                         for j in range(y_int.shape[1]):
                             y_tran_index[i, j] = dict_trans_state[y_tran[j][i]]
@@ -337,7 +341,7 @@ def _om_dist(self, y_int):
         moves_str, unique_indices = np.unique(y_str, axis=0, return_index=True)
         moves_int = y_int[unique_indices]
         uni_num = len(moves_str)
-        dict_move_index = dict(zip(map(tuple, moves_int), range(uni_num)))
+        dict_move_index = dict(zip(map(tuple, moves_int), range(uni_num), strict=False))
 
         # dict_move_index = dict(zip(map(tuple, moves_str), range(uni_num)))
 

diff --git a/giddy/util.py b/giddy/util.py
@@ -179,7 +179,7 @@ def _fill_empty_diagonal_3d(p):
     p0 = p_temp.sum(axis=2) == 0
     if p0.sum() > 0:
         rows, cols = np.where(p0)
-        row_zero_i = list(zip(rows, cols))
+        row_zero_i = list(zip(rows, cols, strict=False))
         for row in row_zero_i:
             i, j = row
             p_temp[i, j, j] = 1