Disable TestComplexity.test_nn_module_test in fbcode (#56677)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56677
This has been failing with `RecursionError: maximum recursion depth
exceeded while calling a Python object` in fbcode for a while now. Obviously
this isn't a fix, but the test works in OSS, so...
ghstack-source-id: 127146338
Test Plan:
```
buck test mode/dev //caffe2/test:jit -- --exact 'caffe2/test:jit - test_nn_module_tests (jit.test_complexity.TestComplexity)' --run-disabled
```
Reviewed By: Lilyjjo
Differential Revision: D27934963
fbshipit-source-id: 21d9858dab9ca1ebb5b67f286e788662dd24a988
diff --git a/test/jit/test_complexity.py b/test/jit/test_complexity.py
index 1a88d12..efea483 100644
--- a/test/jit/test_complexity.py
+++ b/test/jit/test_complexity.py
@@ -1,5 +1,6 @@
import os
import sys
+import unittest
import torch
@@ -12,7 +13,7 @@
from torch.testing._internal.jit_utils import JitTestCase, enable_profiling_mode
from torch.testing._internal.jit_metaprogramming_utils import try_get_nn_module_compiled_mod_and_inputs, \
get_nn_mod_test_name, get_all_nn_module_tests, nn_functional_tests, get_nn_functional_compiled_fn_and_inputs
-from torch.testing._internal.common_utils import run_tests, suppress_warnings
+from torch.testing._internal.common_utils import run_tests, suppress_warnings, IS_FBCODE
def num_ifs_loops(graph):
@@ -66,6 +67,7 @@
print(line)
@suppress_warnings
+ @unittest.skipIf(IS_FBCODE, "Causes a RecursionError in fbcode")
def test_nn_module_tests(self):
with enable_profiling_mode():
stats = [("Name", "Ifs/Loops", "non-tensor ops")]
