numpy-in-pytorch-modules, generic match

trailofbits · Apr 3, 2024 · 3a6867d · 3a6867d
1 parent 0051875
commit 3a6867d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/numpy-in-pytorch-modules.yaml b/python/numpy-in-pytorch-modules.yaml
@@ -17,7 +17,7 @@ rules:
     patterns:
       - pattern-either:
           - pattern: numpy.$FN(...)
-          - pattern: numpy.$MOD.$FN(...)
+          - pattern: numpy. ... .$FN(...)
       - pattern-inside: |
           class $MODULE(torch.nn.Module):
               ...