From 46add4a5c5d9296bece829354efe53a46642cba3 Mon Sep 17 00:00:00 2001 From: ver217 Date: Fri, 18 Mar 2022 13:32:02 +0800 Subject: [PATCH] remove surplus imports --- tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py b/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py index a0667792f..738382d8c 100644 --- a/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py +++ b/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py @@ -4,7 +4,6 @@ from functools import partial import colossalai -import pyte import pytest import torch import torch.distributed as dist