test: remove ds test from p2
@@ -99,15 +99,3 @@ def test_protein_pytorch_geometric() -> None:
exp.run_basic_test_with_temp_config(
config, conf.graphs_examples_path("proteins_pytorch_geometric"), 1
)
-
-
-@pytest.mark.nightly
-def test_deepspeed_cpu_offloading() -> None:
- config = conf.load_config(
- conf.deepspeed_examples_path("cifar10_cpu_offloading/zero_3_cpu_offload.yaml")
- )
- config = conf.set_max_length(config, {"batches": 100})
-
- exp.run_basic_test_with_temp_config(
- config, conf.deepspeed_examples_path("cifar10_cpu_offloading"), 1
- )
