k

firedrakeproject · Nov 16, 2024 · 8417b41 · 8417b41
1 parent cad0cd2
commit 8417b41
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/firedrake/functionspaceimpl.py b/firedrake/functionspaceimpl.py
@@ -195,12 +195,15 @@ def _components(self):
 
     @PETSc.Log.EventDecorator()
     def sub(self, i):
-        #bound = len(self._components)
-        #if i < 0 or i >= bound:
-        #    raise IndexError("Invalid component %d, not in [0, %d)" % (i, bound))
         if len(self) == 1:
+            bound = len(self._components)
+            if i < 0 or i >= bound:
+                raise IndexError("Invalid component %d, not in [0, %d)" % (i, bound))
             return self._components[i]
         else:
+            bound = len(self)
+            if i < 0 or i >= bound:
+                raise IndexError("Invalid index %d, not in [0, %d)" % (i, bound))
             return self.subfunctions[i]
 
     @utils.cached_property