fix stub generation for core sub modules

KratosMultiphysics · Sep 5, 2023 · c21d232 · c21d232
1 parent c3ed8fd
commit c21d232
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: