BUCK for tests (#76)

Summary:

automation

Reviewed By: felipemello1

Differential Revision: D86159099

Author: casteryh

tests/test_resharding_ext.py

@@ -7,13 +7,29 @@
 from logging import getLogger
 
 import pytest
+<<<<<<< dest:   48073f95a923 - yuxuanh: [torchstore] fix multiple tests. with...
+||||||| base:   cfdab62d4cf0 - yuxuanh: [torchstore] fix multiple tests. with...
+
+=======
+import os
+
+>>>>>>> source: de5869157b5c - yuxuanh: [torchstore] BUCK for tests
 from torch.distributed._tensor import Shard
 
 from .test_resharding_basic import _test_resharding
 from .utils import main, transport_plus_strategy_params
 
 logger = getLogger(__name__)
 
+import pytest
+
+def slow_tests_enabled():
+    return os.environ.get("TORCHSTORE_ENABLE_SLOW_TESTS", "0") == "1"
+
+requires_slow_tests_enabled = pytest.mark.skipif(
+    not slow_tests_enabled(),
+    reason="Slow tests are disabled by default, use TORCHSTORE_ENABLE_SLOW_TESTS=1 to enable them",
+)
 
 @pytest.mark.parametrize(*transport_plus_strategy_params())
 @pytest.mark.parametrize(
@@ -53,6 +69,7 @@ async def test_1d_resharding(
     )
 
 
+@requires_slow_tests_enabled
 @pytest.mark.parametrize(*transport_plus_strategy_params())
 @pytest.mark.asyncio
 async def test_2d_to_2d_resharding(strategy_params, use_rdma):