k

Author: ksagiyam

ufl/algorithms/apply_derivatives.py

@@ -207,7 +207,7 @@ def process(self, o: Expr) -> Expr:
             Processed object.
 
         """
-        raise AssertionError(f"UFL expression expected: got {o}")
+        return super().process(o)
 
     @process.register
     def _(self, o: Expr) -> Expr:
@@ -1785,7 +1785,7 @@ def process(self, o: Expr) -> Expr:
             Processed object.
 
         """
-        raise AssertionError(f"UFL expression expected: got {o}")
+        return super().process(o)
 
     @process.register
     def _(self, o: Union[Expr, BaseForm]) -> Union[Expr, BaseForm]:
@@ -2223,7 +2223,7 @@ def process(self, o: Expr) -> Expr:
             Processed object.
 
         """
-        raise AssertionError(f"UFL expression expected: got {o}")
+        return super().process(o)
 
     @process.register
     def _(self, o: Union[Expr, BaseForm]) -> Union[Expr, BaseForm]:

ufl/corealg/dag_traverser.py

@@ -1,13 +1,12 @@
 """Base class for dag traversers."""
 
-from abc import ABC, abstractmethod
 from functools import singledispatchmethod, wraps
 from typing import Union
 
 from ufl.classes import Expr
 
 
-class DAGTraverser(ABC):
+class DAGTraverser:
     """Base class for DAG traversers.
 
     Args:
@@ -59,7 +58,6 @@ def __call__(self, node: Expr, *args) -> Expr:
             return result
 
     @singledispatchmethod
-    @abstractmethod
     def process(self, node: Expr, *args) -> Expr:
         """Process node by type.
 
@@ -71,6 +69,7 @@ def process(self, node: Expr, *args) -> Expr:
             Processed `Expr`.
 
         """
+        raise AssertionError(f"UFL expression expected: got {o}")
 
     def reuse_if_untouched(self, o: Expr) -> Expr:
         """Reuse if touched.