Skip to content

Commit 6133c24

Browse files
Update jax/_src/aot_util.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 98aa196 commit 6133c24

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

jax/_src/aot_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def __hash__(self):
4343
return hash(self.user_key)
4444

4545
def __eq__(self, other):
46-
return hash(self) == hash(other)
46+
return isinstance(other, ComponentKey) and self.user_key == other.user_key
4747

4848
def __str__(self):
4949
return self.user_key

0 commit comments

Comments
 (0)