remove ic 2

mosaicml · Sep 30, 2024 · 4c5802a · 4c5802a
1 parent a6df71f
commit 4c5802a
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/tests/tp/test_tp_strategies.py b/tests/tp/test_tp_strategies.py
@@ -8,7 +8,6 @@
 
 import pytest
 from composer.utils import dist
-from icecream import install
 from omegaconf import OmegaConf as om
 from torch.distributed._tensor import Replicate, Shard
 from torch.distributed.tensor.parallel import (
@@ -23,8 +22,6 @@
 from llmfoundry.utils.config_utils import process_init_device
 from tests.data_utils import create_c4_dataset_xxsmall, gpt_tiny_cfg
 
-install()
-
 
 @pytest.mark.gpu
 @pytest.mark.filterwarnings(