Merge pull request #11539 from KratosMultiphysics/core/post_processin…

…g/stug_generation [Core] Fix stub generation for core sub modules
KratosMultiphysics · Sep 5, 2023 · 5b8bfda · 5b8bfda
2 parents c3ed8fd + c21d232
commit 5b8bfda
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/post_install/stub_generation.py b/scripts/post_install/stub_generation.py
@@ -179,6 +179,7 @@ def PostProcessGeneratedStubFiles():
             data = file_input.read()
 
         data = re.sub(R"import +Kratos(\w+)\n", R"import KratosMultiphysics.\1 as Kratos\1\n", data)
+        data = re.sub(R"import +Kratos.(\w+)\n", R"import KratosMultiphysics.\1\n", data)
         data = data.replace("import Kratos\n", "import KratosMultiphysics as Kratos\n")
 
         with open(str(file.absolute()), "w") as file_output: