Update to new Plum

wesselb · Jan 20, 2024 · 3f064bb · 3f064bb
1 parent 8f04edc
commit 3f064bb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/neuralprocesses/coding.py b/neuralprocesses/coding.py
@@ -31,10 +31,10 @@ def code(coder, xz, z, x, **kw_args):
     """
     if any(
         [
-            isinstance(coder, s.types[0])
-            and issubclass(s.types[0], object)
-            and not issubclass(object, s.types[0])
-            for s in code.methods
+            isinstance(coder, m.signature.types[0])
+            and issubclass(m.signature.types[0], object)
+            and not issubclass(object, m.signature.types[0])
+            for m in code.methods
         ]
     ):
         raise RuntimeError(

diff --git a/pyproject.toml b/pyproject.toml
@@ -17,7 +17,7 @@ dependencies = [
     "numpy>=1.16",
     "backends>=1.4.31",
     "backends-matrix>=1.2.10",
-    "plum-dispatch>=2.2.2",
+    "plum-dispatch>=2.3.0",
     "stheno>=1.3.10",
     "wbml>=0.3.18",
 ]