remove icecream

tests/algorithms/test_algorithms_train.py

@@ -6,8 +6,7 @@
 from composer import Algorithm, Trainer
 from composer.algorithms import GyroDropout, LayerFreezing
 from tests.algorithms.algorithm_settings import get_alg_dataloader, get_alg_kwargs, get_alg_model, get_algs_with_marks
-from icecream import install
-install()
+
 
 @pytest.mark.gpu
 @pytest.mark.parametrize('alg_cls', get_algs_with_marks())
@@ -16,7 +15,6 @@ def test_algorithm_trains(alg_cls: type[Algorithm]):
     alg_kwargs = get_alg_kwargs(alg_cls)
     model = get_alg_model(alg_cls)
     dataloader = get_alg_dataloader(alg_cls)
-    ic(model, dataloader)
     trainer = Trainer(
         model=model,
         train_dataloader=dataloader,

tests/checkpoint/test_state_dict.py

@@ -21,8 +21,7 @@
 from tests.common.compare import deep_compare
 from tests.common.markers import world_size
 from tests.common.models import EvenSimplerMLP, SimpleComposerMLP, configure_tiny_gpt2_hf_model
-from icecream import install
-install()
+
 
 @pytest.mark.gpu
 @pytest.mark.parametrize('use_composer_model', [True, False])

tests/common/models.py

@@ -480,7 +480,6 @@ def loss(self, outputs: torch.Tensor, batch: tuple[Any, torch.Tensor], *args, **
 
     def update_metric(self, batch: Any, outputs: Any, metric: Metric) -> None:
         _, targets = batch
-        ic(metric, outputs, targets)
         metric.update(outputs.squeeze(dim=0), targets.squeeze(dim=0))
 
     def forward(self, batch: tuple[torch.Tensor, Any]) -> torch.Tensor: