mindspore 2.7.0rc1__cp310-cp310-win_amd64.whl → 2.7.1__cp310-cp310-win_amd64.whl

mindspore/parallel/mpi/_mpi_config.py

@@ -1,116 +0,0 @@
-# Copyright 2023 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-# ============================================================================
-"""
-The MPI config, used to configure the MPI environment.
-"""
-import threading
-from mindspore._c_expression import MpiConfig
-from mindspore._checkparam import args_type_check
-
-
-class _MpiConfig:
-    """
-    _MpiConfig is the config tool for controlling MPI
-
-    Note:
-        Create a config through instantiating MpiConfig object is not recommended.
-        should use MpiConfig() to get the config since MpiConfig is singleton.
-    """
-    _instance = None
-    _instance_lock = threading.Lock()
-
-    def __init__(self):
-        self._mpiconfig_handle = MpiConfig.get_instance()
-
-    def __new__(cls, *args, **kwargs):
-        if cls._instance is None:
-            cls._instance_lock.acquire()
-            cls._instance = object.__new__(cls)
-            cls._instance_lock.release()
-        return cls._instance
-
-    def __getattribute__(self, attr):
-        value = object.__getattribute__(self, attr)
-        if attr == "_mpiconfig_handle" and value is None:
-            raise ValueError("mpiconfig handle is none in MpiConfig!!!")
-        return value
-
-    @property
-    def enable_mpi(self):
-        """Get enable mpi."""
-        return self._mpiconfig_handle.get_enable_mpi()
-
-    @enable_mpi.setter
-    def enable_mpi(self, enable_mpi):
-        self._mpiconfig_handle.set_enable_mpi(enable_mpi)
-
-_k_mpi_config = None
-
-
-def _mpi_config():
-    """
-    Get the global mpi config, if mpi config is not created, create a new one.
-
-    Returns:
-        _MpiConfig, the global mpi config.
-    """
-    global _k_mpi_config
-    if _k_mpi_config is None:
-        _k_mpi_config = _MpiConfig()
-    return _k_mpi_config
-
-
-@args_type_check(enable_mpi=bool)
-def _set_mpi_config(**kwargs):
-    """
-    Sets mpi config for running environment.
-
-    mpi config should be configured before running your program. If there is no configuration,
-    mpi module will be disabled by default.
-
-    Note:
-        Attribute name is required for setting attributes.
-
-    Args:
-        enable_mpi (bool): Whether to enable mpi. Default: False.
-
-    Raises:
-        ValueError: If input key is not an attribute in mpi config.
-
-    Examples:
-        >>> mpiconfig.set_mpi_config(enable_mpi=True)
-    """
-    for key, value in kwargs.items():
-        if not hasattr(_mpi_config(), key):
-            raise ValueError("Set mpi config keyword %s is not recognized!" % key)
-        setattr(_mpi_config(), key, value)
-
-
-def _get_mpi_config(attr_key):
-    """
-    Gets mpi config attribute value according to the input key.
-
-    Args:
-        attr_key (str): The key of the attribute.
-
-    Returns:
-        Object, The value of given attribute key.
-
-    Raises:
-        ValueError: If input key is not an attribute in config.
-    """
-    if not hasattr(_mpi_config(), attr_key):
-        raise ValueError("Get context keyword %s is not recognized!" % attr_key)
-    return getattr(_mpi_config(), attr_key)

mindspore/profiler/common/validator/validate_path.py

@@ -1,84 +0,0 @@
-# Copyright 2019 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-# ============================================================================
-"""Validate the input path."""
-import os
-import re
-
-
-def check_valid_character_of_path(file_path):
-    """
-    Validates path.
-
-    The output path of profiler only supports alphabets(a-zA-Z), digit(0-9) or {'-', '_', '.', '/'}.
-
-    Note:
-        Chinese and other paths are not supported at present.
-
-    Args:
-        path (str):  Normalized Path.
-
-    Returns:
-        bool, whether valid.
-    """
-    re_path = r'^[/\\_a-zA-Z0-9-_.@]+$'
-    path_valid = re.fullmatch(re_path, file_path)
-    if not path_valid:
-        msg = "The output path of profiler only supports alphabets(a-zA-Z), " \
-              "digit(0-9) or {'-', '_', '.', '/', '@'}, but got the absolute path= " + file_path
-        raise RuntimeError(msg)
-
-
-def validate_and_normalize_path(
-        path,
-        check_absolute_path=False,
-        allow_parent_dir=True,
-):
-    """
-    Validates path and returns its normalized form.
-
-    If path has a valid scheme, treat path as url, otherwise consider path a
-    unix local path.
-
-    Note:
-        File scheme (rfc8089) is currently not supported.
-
-    Args:
-        path (str): Path to be normalized.
-        check_absolute_path (bool): Whether check path scheme is supported.
-        allow_parent_dir (bool): Whether allow parent dir in path.
-
-    Returns:
-        str, normalized path.
-    """
-    if not path:
-        raise RuntimeError("The path is invalid!")
-
-    path_str = str(path)
-    if not allow_parent_dir:
-        path_components = path_str.split("/")
-        if ".." in path_components:
-            raise RuntimeError("The parent path is not allowed!")
-
-    # path does not have valid schema, treat it as unix local path.
-    if check_absolute_path:
-        if not path_str.startswith("/"):
-            raise RuntimeError("The path is invalid!")
-    try:
-        # most unix systems allow
-        normalized_path = os.path.realpath(path)
-    except ValueError as err:
-        raise RuntimeError("The path is invalid!") from err
-    check_valid_character_of_path(normalized_path)
-    return normalized_path

mindspore/train/memory_profiling_pb2.py

@@ -1,298 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: memory_profiling.proto
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
-from google.protobuf import symbol_database as _symbol_database
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor.FileDescriptor(
-  name='memory_profiling.proto',
-  package='mindspore.profiler',
-  syntax='proto3',
-  serialized_options=None,
-  create_key=_descriptor._internal_create_key,
-  serialized_pb=b'\n\x16memory_profiling.proto\x12\x12mindspore.profiler\"V\n\x0bMemoryProto\x12\x34\n\tgraph_mem\x18\x01 \x03(\x0b\x32!.mindspore.profiler.GraphMemProto\x12\x11\n\ttotal_mem\x18\x02 \x01(\x04\"\xc5\x01\n\rGraphMemProto\x12\x10\n\x08graph_id\x18\x01 \x01(\x03\x12\x12\n\nstatic_mem\x18\x02 \x01(\x03\x12\x33\n\tnode_mems\x18\x03 \x03(\x0b\x32 .mindspore.profiler.NodeMemProto\x12\x37\n\x0btensor_mems\x18\x04 \x03(\x0b\x32\".mindspore.profiler.TensorMemProto\x12\x10\n\x08\x66p_start\x18\x05 \x01(\t\x12\x0e\n\x06\x62p_end\x18\x06 \x01(\t\"\x82\x01\n\x0cNodeMemProto\x12\x11\n\tnode_name\x18\x01 \x01(\t\x12\x0f\n\x07node_id\x18\x02 \x01(\x04\x12\x17\n\x0finput_tensor_id\x18\x03 \x03(\x04\x12\x18\n\x10output_tensor_id\x18\x04 \x03(\x04\x12\x1b\n\x13workspace_tensor_id\x18\x05 \x03(\x04\"x\n\x0eTensorMemProto\x12\x11\n\ttensor_id\x18\x01 \x01(\x04\x12\x0c\n\x04size\x18\x02 \x01(\x04\x12\x0c\n\x04type\x18\x03 \x01(\t\x12\x12\n\nlife_start\x18\x04 \x01(\x04\x12\x10\n\x08life_end\x18\x05 \x01(\x04\x12\x11\n\tlife_long\x18\x06 \x01(\tb\x06proto3'
-)
-
-
-
-
-_MEMORYPROTO = _descriptor.Descriptor(
-  name='MemoryProto',
-  full_name='mindspore.profiler.MemoryProto',
-  filename=None,
-  file=DESCRIPTOR,
-  containing_type=None,
-  create_key=_descriptor._internal_create_key,
-  fields=[
-    _descriptor.FieldDescriptor(
-      name='graph_mem', full_name='mindspore.profiler.MemoryProto.graph_mem', index=0,
-      number=1, type=11, cpp_type=10, label=3,
-      has_default_value=False, default_value=[],
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='total_mem', full_name='mindspore.profiler.MemoryProto.total_mem', index=1,
-      number=2, type=4, cpp_type=4, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-  ],
-  extensions=[
-  ],
-  nested_types=[],
-  enum_types=[
-  ],
-  serialized_options=None,
-  is_extendable=False,
-  syntax='proto3',
-  extension_ranges=[],
-  oneofs=[
-  ],
-  serialized_start=46,
-  serialized_end=132,
-)
-
-
-_GRAPHMEMPROTO = _descriptor.Descriptor(
-  name='GraphMemProto',
-  full_name='mindspore.profiler.GraphMemProto',
-  filename=None,
-  file=DESCRIPTOR,
-  containing_type=None,
-  create_key=_descriptor._internal_create_key,
-  fields=[
-    _descriptor.FieldDescriptor(
-      name='graph_id', full_name='mindspore.profiler.GraphMemProto.graph_id', index=0,
-      number=1, type=3, cpp_type=2, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='static_mem', full_name='mindspore.profiler.GraphMemProto.static_mem', index=1,
-      number=2, type=3, cpp_type=2, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='node_mems', full_name='mindspore.profiler.GraphMemProto.node_mems', index=2,
-      number=3, type=11, cpp_type=10, label=3,
-      has_default_value=False, default_value=[],
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='tensor_mems', full_name='mindspore.profiler.GraphMemProto.tensor_mems', index=3,
-      number=4, type=11, cpp_type=10, label=3,
-      has_default_value=False, default_value=[],
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='fp_start', full_name='mindspore.profiler.GraphMemProto.fp_start', index=4,
-      number=5, type=9, cpp_type=9, label=1,
-      has_default_value=False, default_value=b"".decode('utf-8'),
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='bp_end', full_name='mindspore.profiler.GraphMemProto.bp_end', index=5,
-      number=6, type=9, cpp_type=9, label=1,
-      has_default_value=False, default_value=b"".decode('utf-8'),
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-  ],
-  extensions=[
-  ],
-  nested_types=[],
-  enum_types=[
-  ],
-  serialized_options=None,
-  is_extendable=False,
-  syntax='proto3',
-  extension_ranges=[],
-  oneofs=[
-  ],
-  serialized_start=135,
-  serialized_end=332,
-)
-
-
-_NODEMEMPROTO = _descriptor.Descriptor(
-  name='NodeMemProto',
-  full_name='mindspore.profiler.NodeMemProto',
-  filename=None,
-  file=DESCRIPTOR,
-  containing_type=None,
-  create_key=_descriptor._internal_create_key,
-  fields=[
-    _descriptor.FieldDescriptor(
-      name='node_name', full_name='mindspore.profiler.NodeMemProto.node_name', index=0,
-      number=1, type=9, cpp_type=9, label=1,
-      has_default_value=False, default_value=b"".decode('utf-8'),
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='node_id', full_name='mindspore.profiler.NodeMemProto.node_id', index=1,
-      number=2, type=4, cpp_type=4, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='input_tensor_id', full_name='mindspore.profiler.NodeMemProto.input_tensor_id', index=2,
-      number=3, type=4, cpp_type=4, label=3,
-      has_default_value=False, default_value=[],
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='output_tensor_id', full_name='mindspore.profiler.NodeMemProto.output_tensor_id', index=3,
-      number=4, type=4, cpp_type=4, label=3,
-      has_default_value=False, default_value=[],
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='workspace_tensor_id', full_name='mindspore.profiler.NodeMemProto.workspace_tensor_id', index=4,
-      number=5, type=4, cpp_type=4, label=3,
-      has_default_value=False, default_value=[],
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-  ],
-  extensions=[
-  ],
-  nested_types=[],
-  enum_types=[
-  ],
-  serialized_options=None,
-  is_extendable=False,
-  syntax='proto3',
-  extension_ranges=[],
-  oneofs=[
-  ],
-  serialized_start=335,
-  serialized_end=465,
-)
-
-
-_TENSORMEMPROTO = _descriptor.Descriptor(
-  name='TensorMemProto',
-  full_name='mindspore.profiler.TensorMemProto',
-  filename=None,
-  file=DESCRIPTOR,
-  containing_type=None,
-  create_key=_descriptor._internal_create_key,
-  fields=[
-    _descriptor.FieldDescriptor(
-      name='tensor_id', full_name='mindspore.profiler.TensorMemProto.tensor_id', index=0,
-      number=1, type=4, cpp_type=4, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='size', full_name='mindspore.profiler.TensorMemProto.size', index=1,
-      number=2, type=4, cpp_type=4, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='type', full_name='mindspore.profiler.TensorMemProto.type', index=2,
-      number=3, type=9, cpp_type=9, label=1,
-      has_default_value=False, default_value=b"".decode('utf-8'),
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='life_start', full_name='mindspore.profiler.TensorMemProto.life_start', index=3,
-      number=4, type=4, cpp_type=4, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='life_end', full_name='mindspore.profiler.TensorMemProto.life_end', index=4,
-      number=5, type=4, cpp_type=4, label=1,
-      has_default_value=False, default_value=0,
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-    _descriptor.FieldDescriptor(
-      name='life_long', full_name='mindspore.profiler.TensorMemProto.life_long', index=5,
-      number=6, type=9, cpp_type=9, label=1,
-      has_default_value=False, default_value=b"".decode('utf-8'),
-      message_type=None, enum_type=None, containing_type=None,
-      is_extension=False, extension_scope=None,
-      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
-  ],
-  extensions=[
-  ],
-  nested_types=[],
-  enum_types=[
-  ],
-  serialized_options=None,
-  is_extendable=False,
-  syntax='proto3',
-  extension_ranges=[],
-  oneofs=[
-  ],
-  serialized_start=467,
-  serialized_end=587,
-)
-
-_MEMORYPROTO.fields_by_name['graph_mem'].message_type = _GRAPHMEMPROTO
-_GRAPHMEMPROTO.fields_by_name['node_mems'].message_type = _NODEMEMPROTO
-_GRAPHMEMPROTO.fields_by_name['tensor_mems'].message_type = _TENSORMEMPROTO
-DESCRIPTOR.message_types_by_name['MemoryProto'] = _MEMORYPROTO
-DESCRIPTOR.message_types_by_name['GraphMemProto'] = _GRAPHMEMPROTO
-DESCRIPTOR.message_types_by_name['NodeMemProto'] = _NODEMEMPROTO
-DESCRIPTOR.message_types_by_name['TensorMemProto'] = _TENSORMEMPROTO
-_sym_db.RegisterFileDescriptor(DESCRIPTOR)
-
-MemoryProto = _reflection.GeneratedProtocolMessageType('MemoryProto', (_message.Message,), {
-  'DESCRIPTOR' : _MEMORYPROTO,
-  '__module__' : 'memory_profiling_pb2'
-  # @@protoc_insertion_point(class_scope:mindspore.profiler.MemoryProto)
-  })
-_sym_db.RegisterMessage(MemoryProto)
-
-GraphMemProto = _reflection.GeneratedProtocolMessageType('GraphMemProto', (_message.Message,), {
-  'DESCRIPTOR' : _GRAPHMEMPROTO,
-  '__module__' : 'memory_profiling_pb2'
-  # @@protoc_insertion_point(class_scope:mindspore.profiler.GraphMemProto)
-  })
-_sym_db.RegisterMessage(GraphMemProto)
-
-NodeMemProto = _reflection.GeneratedProtocolMessageType('NodeMemProto', (_message.Message,), {
-  'DESCRIPTOR' : _NODEMEMPROTO,
-  '__module__' : 'memory_profiling_pb2'
-  # @@protoc_insertion_point(class_scope:mindspore.profiler.NodeMemProto)
-  })
-_sym_db.RegisterMessage(NodeMemProto)
-
-TensorMemProto = _reflection.GeneratedProtocolMessageType('TensorMemProto', (_message.Message,), {
-  'DESCRIPTOR' : _TENSORMEMPROTO,
-  '__module__' : 'memory_profiling_pb2'
-  # @@protoc_insertion_point(class_scope:mindspore.profiler.TensorMemProto)
-  })
-_sym_db.RegisterMessage(TensorMemProto)
-
-
-# @@protoc_insertion_point(module_scope)

mindspore/utils/hooks.py

@@ -1,81 +0,0 @@
-# Copyright 2025 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# 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.
-# ============================================================================
-"""hooks"""
-from collections import OrderedDict
-import weakref
-from typing import Any, Tuple
-
-
-class _RemovableHandle:
-    r"""
-    A handle which provides the capability to remove a hook.
-
-    Args:
-        hooks_dict (dict): A dictionary of hooks, indexed by hook `id`.
-
-    Keyword Args:
-        extra_dict (Union[dict, list[dict]], optional): An additional dictionary or list of
-            dictionaries whose keys will be deleted when the same keys are
-            removed from `hooks_dict`. Default ``None``.
-    """
-
-    id: int
-    next_id: int = 0
-
-    def __init__(self, hooks_dict: Any, *, extra_dict: Any = None) -> None:
-        self.hooks_dict_ref = weakref.ref(hooks_dict)
-        self.id = _RemovableHandle.next_id
-        _RemovableHandle.next_id += 1
-
-        self.extra_dict_ref: Tuple = ()
-        if isinstance(extra_dict, dict):
-            self.extra_dict_ref = (weakref.ref(extra_dict),)
-        elif isinstance(extra_dict, list):
-            self.extra_dict_ref = tuple(weakref.ref(d) for d in extra_dict)
-
-    def remove(self) -> None:
-        hooks_dict = self.hooks_dict_ref()
-        if hooks_dict is not None and self.id in hooks_dict:
-            del hooks_dict[self.id]
-
-        for ref in self.extra_dict_ref:
-            extra_dict = ref()
-            if extra_dict is not None and self.id in extra_dict:
-                del extra_dict[self.id]
-
-    def __getstate__(self):
-        if self.extra_dict_ref is None:
-            return (self.hooks_dict_ref(), self.id)
-        return (self.hooks_dict_ref(), self.id, tuple(ref() for ref in self.extra_dict_ref))
-
-    def __setstate__(self, state) -> None:
-        if state[0] is None:
-            # create a dead reference
-            self.hooks_dict_ref = weakref.ref(OrderedDict())
-        else:
-            self.hooks_dict_ref = weakref.ref(state[0])
-        self.id = state[1]
-        _RemovableHandle.next_id = max(_RemovableHandle.next_id, self.id + 1)
-
-        if len(state) < 3 or state[2] is None:
-            self.extra_dict_ref = ()
-        else:
-            self.extra_dict_ref = tuple(weakref.ref(d) for d in state[2])
-
-    def __enter__(self) -> "_RemovableHandle":
-        return self
-
-    def __exit__(self, type: Any, value: Any, tb: Any) -> None:
-        self.remove()