Revert back future Plum update

wesselb · Nov 8, 2023 · 8f04edc · 8f04edc
1 parent 2508919
commit 8f04edc
Showing 1 changed file with 4 additions and 4 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, method.signature.types[0])
-            and issubclass(method.signature.types[0], object)
-            and not issubclass(object, method.signature.types[0])
-            for method in code.methods
+            isinstance(coder, s.types[0])
+            and issubclass(s.types[0], object)
+            and not issubclass(object, s.types[0])
+            for s in code.methods
         ]
     ):
         raise RuntimeError(