SONARPY-2148: Add pytorch and torchvision custom stubs

Author: Seppli11

python-checks/src/main/java/org/sonar/python/checks/TorchModuleShouldCallInitCheck.java

@@ -41,7 +41,7 @@
 
 @Rule(key = "S6978")
 public class TorchModuleShouldCallInitCheck extends PythonSubscriptionCheck {
-  private static final String TORCH_NN_MODULE = "torch.nn.Module";
+  private static final String TORCH_NN_MODULE = "torch.nn.modules.module.Module";
   private static final String MESSAGE = "Add a call to super().__init__().";
   private static final String SECONDARY_MESSAGE = "Inheritance happens here";
   public static final String QUICK_FIX_MESSAGE = "insert call to super constructor";

python-checks/src/test/java/org/sonar/python/checks/TorchModuleShouldCallInitCheckTest.java

@@ -33,13 +33,13 @@ void test() {
   void testQuickFix() {
     PythonQuickFixVerifier.verify(new TorchModuleShouldCallInitCheck(),
       """
-        import torch
+        import torch.nn
         class Test(torch.nn.Module):
           def __init__(self):
             some_method()
         """,
       """
-        import torch
+        import torch.nn
         class Test(torch.nn.Module):
           def __init__(self):
             super().__init__()
@@ -48,13 +48,13 @@ def __init__(self):
 
     PythonQuickFixVerifier.verify(new TorchModuleShouldCallInitCheck(),
       """
-        import torch
+        import torch.nn
         class Test(torch.nn.Module):
           def __init__(self):
             ... 
         """,
       """
-        import torch
+        import torch.nn
         class Test(torch.nn.Module):
           def __init__(self):
             super().__init__()
@@ -63,14 +63,14 @@ def __init__(self):
 
     PythonQuickFixVerifier.verifyNoQuickFixes(new TorchModuleShouldCallInitCheck(),
       """
-        import torch
+        import torch.nn
         class Test(torch.nn.Module):
           def __init__(self): some_method()
         """);
 
     PythonQuickFixVerifier.verifyNoQuickFixes(new TorchModuleShouldCallInitCheck(),
       """
-        import torch
+        import torch.nn
         class Test(torch.nn.Module):
           def __init__(self): pass
         """);