Fix test ownership lint (#71554)

Summary:
I noticed after creating https://github.com/pytorch/pytorch/issues/71553 that the test ownership lint was not working properly.

This fixes my egregious mistake and fixes the broken lints.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/71554

Reviewed By: malfet

Differential Revision: D33690732

Pulled By: janeyx99

fbshipit-source-id: ba4dfbcd98038e4afd63e326832ae40935d2501e
(cherry picked from commit 1bbc3d343a)

.github/scripts/lint_test_ownership.py

@@ -22,7 +22,7 @@ from typing import List, Any
 # Team/owner labels usually start with "module: " or "oncall: ", but the following are acceptable exceptions
 ACCEPTABLE_OWNER_LABELS = ["NNC", "high priority"]
 GLOB_EXCEPTIONS = [
-    "test/run_test.py"
+    "**/test/run_test.py"
 ]
 
 PYTORCH_ROOT = Path(__file__).resolve().parent.parent.parent
@@ -35,7 +35,7 @@ S3_RESOURCE_READ_ONLY = boto3.resource("s3", config=botocore.config.Config(signa
 def get_all_test_files() -> List[Path]:
     test_files = list(TEST_DIR.glob("**/test_*.py"))
     test_files.extend(list(TEST_DIR.glob("**/*_test.py")))
-    return [f for f in test_files if any([fnmatch.fnmatch(str(f), g) for g in GLOB_EXCEPTIONS])]
+    return [f for f in test_files if not any([fnmatch.fnmatch(str(f), g) for g in GLOB_EXCEPTIONS])]
 
 
 def get_pytorch_labels() -> Any:

test/fx2trt/converters/acc_op/test_adaptive_avgpool.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_avgpool.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_batchnorm.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_binary_ops.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 from typing import Callable
 

test/fx2trt/converters/acc_op/test_cat.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_chunk.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops
 import torch.nn as nn

test/fx2trt/converters/acc_op/test_clamp.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_convolution.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_dequantize.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import unittest
 

test/fx2trt/converters/acc_op/test_elu.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_flatten.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_gelu.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops
 import torch.nn as nn

test/fx2trt/converters/acc_op/test_getitem.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_hard_sigmoid.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 from torch.testing._internal.common_fx2trt import InputTensorSpec, AccTestCase
 from torch import nn
 from torch.fx.experimental.fx_acc import acc_ops

test/fx2trt/converters/acc_op/test_hardtanh.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_layernorm.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_leaky_relu.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_linear.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_matmul.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_max.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_maximum.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_maxpool.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_min.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_minimum.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_narrow.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_pad.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_permute.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_quantize_per_tensor.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import unittest
 

test/fx2trt/converters/acc_op/test_reduce_ops.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_relu.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_reshape.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_selu.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_sigmoid.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_silu.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 from torch.testing._internal.common_fx2trt import InputTensorSpec, AccTestCase
 from torch import nn
 import torch

test/fx2trt/converters/acc_op/test_size.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_softmax.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_softsign.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_split.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_squeeze.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_tanh.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_tile.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_topk.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops

test/fx2trt/converters/acc_op/test_unary_ops.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 from typing import Callable
 

test/fx2trt/converters/acc_op/test_unsqueeze.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx

test/fx2trt/converters/vanilla/test_add.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import operator
 

test/fx2trt/converters/vanilla/test_convolution.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch
 import torch.fx

test/fx2trt/core/test_import_fx2trt.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 # (c) Facebook, Inc. and its affiliates. Confidential and proprietary.
 
+# Owner(s): ["oncall: aiacc"]
 
 # Test that this import should not trigger any error when run
 # in non-GPU hosts, or in any build mode.

test/fx2trt/core/test_input_tensor_spec.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 from typing import List, Optional
 
 import torch

test/fx2trt/core/test_trt_module.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import torch
 import torch.fx
 import torch.fx.experimental.fx_acc.acc_tracer as acc_tracer

test/fx2trt/passes/test_fuse_permute_linear_trt.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase

test/fx2trt/passes/test_fuse_permute_matmul_trt.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase

test/fx2trt/passes/test_fuse_unsqueeze_cat_sum_trt.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase

test/fx2trt/passes/test_multi_fuse_trt.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import torch
 import torch.fx.experimental.fx_acc.acc_ops as acc_ops
 from torch.testing._internal.common_fx2trt import AccTestCase

test/fx2trt/passes/test_remove_duplicate_output_args.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import logging
 
 import torch.fx as fx

test/fx2trt/trt_lower/test_fx2trt_lower.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: aiacc"]
+
 import logging
 # @manual=//caffe2:torch_fx2trt
 from torch.fx.experimental.fx2trt.lower import Lowerer

test/fx2trt/trt_lower/trt_operator_supported_test.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import torch.fx.experimental.fx_acc.acc_ops  # noqa: F401
 import torch

test/fx2trt/trt_lower/trt_splitter_test.py

@@ -1,4 +1,4 @@
-# Owner(s): ["oncall: fx"]
+# Owner(s): ["oncall: aiacc"]
 
 import operator
 

test/jit/test_alias_analysis.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: jit"]
+
 from torch.testing._internal.jit_utils import JitTestCase
 from torch._C import parse_ir
 import torch

test/jit/test_device_analysis.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: jit"]
+
 from itertools import product
 import unittest
 

test/jit/test_dtype_analysis.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: jit"]
+
 from itertools import product
 from typing import Tuple
 from unittest.case import expectedFailure

test/jit/test_if_hoisting.py

@@ -1,3 +1,4 @@
+# Owner(s): ["oncall: jit"]
 
 import torch
 from torch.testing import FileCheck

test/mobile/test_upgraders.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: mobile"]
+
 import torch
 import torch.utils.bundled_inputs
 import io

test/quantization/dbr/test_quantize_dbr.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: quantization"]
+
 import collections
 import copy
 import math

test/quantization/fx/test_subgraph_rewriter.py

@@ -1,3 +1,4 @@
+# Owner(s): ["oncall: quantization"]
 # Copied from pytorch/test/fx/test_subgraph_rewriter.py
 
 import os

test/test_functionalization.py

@@ -1,3 +1,5 @@
+# Owner(s): ["module: codegen"]
+
 import torch
 from torch.testing._internal.common_utils import TestCase, run_tests
 from torch.testing._internal.logging_tensor import LoggingTensor, capture_logs, log_input

test/test_masked.py

@@ -1,3 +1,5 @@
+# Owner(s): ["module: masked operators"]
+
 """Tests for masked operations.
 """
 

test/test_monitor.py

@@ -1,3 +1,5 @@
+# Owner(s): ["oncall: r2p"]
+
 from torch.testing._internal.common_utils import (
     TestCase, run_tests,
 )