| 1234567891011121314151617181920212223 |
- # Copyright (c) Facebook, Inc. and its affiliates.
- # All rights reserved.
- #
- # This source code is licensed under the BSD-style license found in the
- # LICENSE file in the root directory of this source tree.
- import warnings
- # TODO: remove this file when the migration of the pytree utility is done
- from torch.utils._pytree import tree_map_, treespec_pprint
- __all__ = ["tree_map_", "treespec_pprint"]
- with warnings.catch_warnings():
- warnings.simplefilter("always")
- warnings.warn(
- "`torch._functorch.pytree_hacks` is deprecated and will be removed in a future release. "
- "Please `use torch.utils._pytree` instead.",
- DeprecationWarning,
- stacklevel=2,
- )
|
