[Andrew Polyakov] Do not substitute second parameter for NDArray for mypy

Author: AndrewPolyakov1

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/PythonCodeGenerator.kt

@@ -107,8 +107,18 @@ class PythonCodeGenerator(
         imports.forEach { renderer.renderPythonImport(it) }
 
         val paramNames = method.argumentsNames
-        val parameters = paramNames.map { argument ->
-            "${argument}: ${methodAnnotations[argument]?.pythonTypeRepresentation() ?: pythonAnyType.pythonTypeRepresentation()}"
+        val parameters: List<String> = paramNames.map { argument ->
+            if (methodAnnotations[argument]?.meta.toString() == "numpy.ndarray") {
+                val re = """, .*[^]]""".toRegex()
+                val type = re.find(methodAnnotations[argument]?.pythonTypeRepresentation().toString())?.value
+                val newAnnotation: String =
+                    type?.let {
+                        methodAnnotations[argument]?.pythonTypeRepresentation().toString()
+                            .replace(it, ", ${pythonAnyType.pythonTypeRepresentation()}")
+                    } ?: pythonAnyType.pythonTypeRepresentation()
+                "${argument}: $newAnnotation"
+            } else
+                "${argument}: ${methodAnnotations[argument]?.pythonTypeRepresentation() ?: pythonAnyType.pythonTypeRepresentation()}"
         }
 
         val functionPrefix = "__mypy_check"