Handle function when computing cache keys

iiasa · Dec 3, 2024 · f9e8d71 · f9e8d71
1 parent c518740
commit f9e8d71
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/message_ix_models/util/cache.py b/message_ix_models/util/cache.py
@@ -14,7 +14,8 @@
 import logging
 from collections.abc import Callable
 from dataclasses import asdict, is_dataclass
-from typing import TYPE_CHECKING
+from types import FunctionType
+from typing import TYPE_CHECKING, Union
 
 import genno.caching
 import ixmp
@@ -60,6 +61,11 @@ def _dataclass(o: object):
     )
 
 
+@genno.caching.Encoder.register
+def _repr_only(o: Union[FunctionType]):
+    return repr(o)
+
+
 @genno.caching.Encoder.register
 def _si(o: ScenarioInfo):
     return dict(o.set)