Skip to content

Commit a705ad6

Browse files
committed
Performance improvement.
1 parent f160904 commit a705ad6

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

inflation/cliques_with_symmetry.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def find_cliques_symmetric(
128128
# --- B. GENERATE CLIQUE ORBIT ---
129129
# unique_in_orbit, where_unique = np.unique(permuted_bases, axis=0, return_index=True)
130130
# all_found_cliques.extend(unique_in_orbit)
131-
all_found_clique_masks.extend(base_mask[automorphisms])
131+
all_found_clique_masks.extend(np.unique(base_mask[automorphisms], axis=0))
132132

133133

134134

@@ -149,8 +149,9 @@ def find_cliques_symmetric(
149149
new_cnbrs_mask[:u + 1] = False
150150

151151
queue.append((new_base_mask, new_cnbrs_mask))
152-
all_found_clique_masks = np.unique(all_found_clique_masks, axis=0)
153-
return np.unique(all_found_clique_masks, axis=0).astype(bool)
152+
all_found_clique_masks = np.unique(all_found_clique_masks, axis=0).astype(bool)
153+
print("Queue complete.")
154+
return all_found_clique_masks
154155

155156
def all_and_maximal_cliques_symmetry(adjmat: np.ndarray,
156157
symgroup: np.ndarray,
@@ -184,6 +185,8 @@ def all_and_maximal_cliques_symmetry(adjmat: np.ndarray,
184185
max_cliques = np.empty((0, adjmat.shape[0]), dtype=bool)
185186
return (sorted([np.flatnonzero(bm).tolist() for bm in all_cliques], key=len),
186187
sorted([np.flatnonzero(bm).tolist() for bm in max_cliques], key=len))
188+
# return (all_cliques,
189+
# max_cliques)
187190

188191
if __name__ == '__main__':
189192
### Example Usage

0 commit comments

Comments
 (0)