wrap in enum.member

Author: matsumotosan

tests/tests_pytorch/strategies/test_ddp_integration_comm_hook.py

@@ -11,6 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+import sys
+from enum import member
 from unittest import mock
 
 import pytest
@@ -23,6 +25,9 @@
 
 if torch.distributed.is_available():
     import torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook as post_localSGD
+
+    if sys.version_info >= (3, 13):
+        post_localSGD = member(post_localSGD)
     from torch.distributed.algorithms.ddp_comm_hooks import default_hooks as default
     from torch.distributed.algorithms.ddp_comm_hooks import powerSGD_hook as powerSGD