Fix bug in layers.ops.deserialize_scatter (close #347)

danielegrattarola · Mar 11, 2022 · c1d8878 · c1d8878
1 parent 4aaaff5
commit c1d8878
Showing 1 changed file with 6 additions and 12 deletions.
diff --git a/spektral/layers/ops/scatter.py b/spektral/layers/ops/scatter.py
@@ -214,15 +214,9 @@ def serialize_scatter(identifier):
 
 
 def deserialize_scatter(scatter):
-    if isinstance(scatter, str):
-        if scatter in OP_DICT:
-            return OP_DICT[scatter]
-        else:
-            if callable(scatter):
-                return scatter
-            else:
-                raise ValueError(
-                    "scatter must be callable or string in: {}.".format(
-                        list(OP_DICT.keys())
-                    )
-                )
+    if isinstance(scatter, str) and scatter in OP_DICT:
+        return OP_DICT[scatter]
+    elif callable(scatter):
+        return scatter
+    else:
+        raise ValueError(f"scatter must be callable or string: {list(OP_DICT.keys())}")