BUCK for tests (#76)

Summary:

automation

Reviewed By: felipemello1

Differential Revision: D86159099

Author: casteryh

tests/test_resharding_ext.py

@@ -4,6 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+import os
 from logging import getLogger
 
 import pytest
@@ -15,6 +16,16 @@
 logger = getLogger(__name__)
 
 
+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(
     "put_mesh_shape,get_mesh_shape,put_sharding_dim,get_sharding_dim",
@@ -53,6 +64,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):