opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-28 20:43:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support hashing Color type in hash_obj

Browse Source
This commit is contained in:
YishiMichael
2022-04-15 13:48:24 +08:00
parent 020bd87271
commit 09952756ce
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
manimlib/utils/iterables.py
View File

@ -1,5 +1,7 @@
from __future__ import annotations from __future__ import annotations
from colour import Color
import numpy as np import numpy as np
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
@ -79,7 +81,7 @@ def batch_by_property(
return batch_prop_pairs return batch_prop_pairs
def listify(obj) -> list: def listify(obj: object) -> list:
if isinstance(obj, str): if isinstance(obj, str):
return [obj] return [obj]
try: try:
@ -139,4 +141,7 @@ def hash_obj(obj: object) -> int:
if isinstance(obj, (set, tuple, list)): if isinstance(obj, (set, tuple, list)):
return hash(tuple(hash_obj(e) for e in obj)) return hash(tuple(hash_obj(e) for e in obj))
if isinstance(obj, Color):
return hash(obj.get_rgb())
return hash(obj) return hash(obj)