From af06c50553147b95774f673b361b071c7e86a159 Mon Sep 17 00:00:00 2001
From: Marcus Mohr <marcus.mohr@lmu.de>
Date: Tue, 3 Dec 2024 10:40:43 +0100
Subject: [PATCH] Fixes details in function_space_impl_factory.py

Adds missing return type annotation and imports HOGException.
---
 .../function_space_impl_factory.py                            | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/hog/operator_generation/function_space_implementations/function_space_impl_factory.py b/hog/operator_generation/function_space_implementations/function_space_impl_factory.py
index e0d0b16..d3071ff 100644
--- a/hog/operator_generation/function_space_implementations/function_space_impl_factory.py
+++ b/hog/operator_generation/function_space_implementations/function_space_impl_factory.py
@@ -14,6 +14,8 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
+from hog.exception import HOGException
+
 from hog.function_space import (
     FunctionSpace,
     LagrangianFunctionSpace,
@@ -44,7 +46,7 @@ def create_impl(
     name: str,
     type_descriptor: HOGType,
     is_pointer: bool = False,
-):
+) -> type:
     """Takes a mathematical function space and produces the corresponding function space implementation.
 
     :param func_space:      The mathematical function space.
-- 
GitLab