From 986ac03e374a00a52cee98c8ac14fb1ba6b66610 Mon Sep 17 00:00:00 2001 From: Michael Benayoun Date: Wed, 23 Jun 2021 18:16:24 +0200 Subject: [PATCH] changed modeling_fx_utils.py to utils/fx.py for clarity (#12326) Co-authored-by: Michael Benayoun --- src/transformers/{modeling_fx_utils.py => utils/fx.py} | 4 ++-- tests/test_modeling_common.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) rename src/transformers/{modeling_fx_utils.py => utils/fx.py} (99%) diff --git a/src/transformers/modeling_fx_utils.py b/src/transformers/utils/fx.py similarity index 99% rename from src/transformers/modeling_fx_utils.py rename to src/transformers/utils/fx.py index 8e513c811b..6eadfa3fa0 100644 --- a/src/transformers/modeling_fx_utils.py +++ b/src/transformers/utils/fx.py @@ -11,7 +11,7 @@ from torch.fx.node import Argument from transformers.file_utils import TORCH_FX_REQUIRED_VERSION, importlib_metadata, is_torch_fx_available -from . import ( +from .. import ( MODEL_FOR_CAUSAL_LM_MAPPING, MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING, MODEL_FOR_MASKED_LM_MAPPING, @@ -26,7 +26,7 @@ from . import ( PreTrainedModel, logging, ) -from .models.auto import get_values +from ..models.auto import get_values logger = logging.get_logger(__name__) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 42913d1fb2..dbdf45ac8b 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -67,7 +67,7 @@ if is_torch_available(): ) if is_torch_fx_available(): - from transformers.modeling_fx_utils import symbolic_trace + from transformers.utils.fx import symbolic_trace def _config_zero_init(config):