temp

firedrakeproject · Nov 13, 2024 · 526893f · 526893f
1 parent 58a53f1
commit 526893f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/ufl/algorithms/apply_restrictions.py b/ufl/algorithms/apply_restrictions.py
@@ -192,17 +192,15 @@ def apply_restrictions(expression, assume_single_integral_type=True):
         # ``exterior_facet`` and the latter ``interior_facet``.
         integral_types = None
     rules = RestrictionPropagator(assume_single_integral_type=assume_single_integral_type)
-    if isinstance(expression, Form):
-        return map_integrand_dags(rules, expression, only_integral_type=integral_types)
-    elif isinstance(expression, FormData):
+    if isinstance(expression, FormData):
         for integral_data in expression.integral_data:
             integral_data.integrals = tuple(
                 map_integrand_dags(rules, integral, only_integral_type=integral_types)
                 for integral in integral_data.integrals
             )
         return expression
     else:
-        raise NotImplementedError(f"Unable to handle {type(expression)}")
+        return map_integrand_dags(rules, expression, only_integral_type=integral_types)
 
 
 class DefaultRestrictionApplier(MultiFunction):