Browse Source
Use bool to replace np.bool.
Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb
2 years ago
1 changed files with
1 additions and
1 deletions
-
tn.py
|
@ -110,7 +110,7 @@ def tn(sims: np.ndarray, |
|
|
|
|
|
|
|
|
# Constraints C3 |
|
|
# Constraints C3 |
|
|
if len(intermediate_rs) == 0: |
|
|
if len(intermediate_rs) == 0: |
|
|
C3 = np.ones(C2.shape, dtype=np.bool) |
|
|
|
|
|
|
|
|
C3 = np.ones(C2.shape, dtype=bool) |
|
|
else: |
|
|
else: |
|
|
# "the equal sign" in C3 in paper is wrong because it's contradictory to C2 |
|
|
# "the equal sign" in C3 in paper is wrong because it's contradictory to C2 |
|
|
cond1 = intermediate_rs[None, :] > r_i[:, None] |
|
|
cond1 = intermediate_rs[None, :] > r_i[:, None] |
|
|