From f92e983fa4af14147836230c764f885e338cca8b Mon Sep 17 00:00:00 2001
From: Marcus Mohr <marcus.mohr@lmu.de>
Date: Tue, 3 Dec 2024 10:54:31 +0100
Subject: [PATCH] Fixes return type to FunctionSpaceImpl

---
 .../function_space_impl_factory.py                          | 6 +++++-
 1 file changed, 5 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 d3071ff..c3762ee 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
@@ -23,6 +23,10 @@ from hog.function_space import (
     N1E1Space,
 )
 
+from hog.operator_generation.function_space_implementations.function_space_impl_base import (
+    FunctionSpaceImpl,
+)
+
 from hog.operator_generation.function_space_implementations.p0_space_impl import (
     P0FunctionSpaceImpl,
 )
@@ -46,7 +50,7 @@ def create_impl(
     name: str,
     type_descriptor: HOGType,
     is_pointer: bool = False,
-) -> type:
+) -> FunctionSpaceImpl:
     """Takes a mathematical function space and produces the corresponding function space implementation.
 
     :param func_space:      The mathematical function space.
-- 
GitLab