We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d975305 commit d632995Copy full SHA for d632995
tests/tests_pytorch/strategies/test_ddp_integration_comm_hook.py
@@ -11,7 +11,6 @@
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14
-import sys
15
from unittest import mock
16
17
import pytest
@@ -24,11 +23,6 @@
24
23
25
if torch.distributed.is_available():
26
import torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook as post_localSGD
27
-
28
- if sys.version_info >= (3, 13):
29
- from enum import member
30
31
- post_localSGD = member(post_localSGD)
32
from torch.distributed.algorithms.ddp_comm_hooks import default_hooks as default
33
from torch.distributed.algorithms.ddp_comm_hooks import powerSGD_hook as powerSGD
34
0 commit comments