gh-94773: deepfreeze: support frozensets with unsortable types (GH-94775)

This commit is contained in:
Christian Heimes 2022-07-12 18:09:47 +02:00 committed by GitHub
parent ec5db539b9
commit 0c66074e9f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 1 deletions

View file

@ -0,0 +1,2 @@
``deepfreeze.py`` now supports code object with frozensets that contain
incompatible, unsortable types.

View file

@ -359,7 +359,12 @@ def generate_complex(self, name: str, z: complex) -> str:
return f"&{name}.ob_base"
def generate_frozenset(self, name: str, fs: FrozenSet[object]) -> str:
ret = self.generate_tuple(name, tuple(sorted(fs)))
try:
fs = sorted(fs)
except TypeError:
# frozen set with incompatible types, fallback to repr()
fs = sorted(fs, key=repr)
ret = self.generate_tuple(name, tuple(fs))
self.write("// TODO: The above tuple should be a frozenset")
return ret