[Refactor] harmonize the module structure for models in tests (huggin…

…gface#6738) * harmonize the module structure for models in tests * make the folders modules. --------- Co-authored-by: YiYi Xu <yixu310@gmail.com>
nileshkokane01 · Feb 1, 2024 · ec9840a · ec9840a
1 parent 093a03a
commit ec9840a
Show file tree

Hide file tree

Showing 15 changed files with 10 additions and 10 deletions.
diff --git a/tests/models/autoencoders/__init__.py b/tests/models/autoencoders/__init__.py
diff --git a/tests/models/test_models_vae.py → tests/models/autoencoders/test_models_vae.py b/tests/models/test_models_vae.py → tests/models/autoencoders/test_models_vae.py
@@ -46,7 +46,7 @@
 )
 from diffusers.utils.torch_utils import randn_tensor
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 enable_full_determinism()

diff --git a/tests/models/test_models_vae_flax.py → ...dels/autoencoders/test_models_vae_flax.py b/tests/models/test_models_vae_flax.py → ...dels/autoencoders/test_models_vae_flax.py
@@ -4,7 +4,7 @@
 from diffusers.utils import is_flax_available
 from diffusers.utils.testing_utils import require_flax
 
-from .test_modeling_common_flax import FlaxModelTesterMixin
+from ..test_modeling_common_flax import FlaxModelTesterMixin
 
 
 if is_flax_available():

diff --git a/tests/models/test_models_vq.py → tests/models/autoencoders/test_models_vq.py b/tests/models/test_models_vq.py → tests/models/autoencoders/test_models_vq.py
@@ -25,7 +25,7 @@
     torch_device,
 )
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 enable_full_determinism()

diff --git a/tests/models/unets/__init__.py b/tests/models/unets/__init__.py
diff --git a/tests/models/test_models_unet_1d.py → tests/models/unets/test_models_unet_1d.py b/tests/models/test_models_unet_1d.py → tests/models/unets/test_models_unet_1d.py
@@ -25,7 +25,7 @@
     torch_device,
 )
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 class UNet1DModelTests(ModelTesterMixin, UNetTesterMixin, unittest.TestCase):

diff --git a/tests/models/test_models_unet_2d.py → tests/models/unets/test_models_unet_2d.py b/tests/models/test_models_unet_2d.py → tests/models/unets/test_models_unet_2d.py
@@ -30,7 +30,7 @@
     torch_device,
 )
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 logger = logging.get_logger(__name__)

diff --git a/...s/models/test_models_unet_2d_condition.py → ...ls/unets/test_models_unet_2d_condition.py b/...s/models/test_models_unet_2d_condition.py → ...ls/unets/test_models_unet_2d_condition.py
@@ -48,7 +48,7 @@
     torch_device,
 )
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 logger = logging.get_logger(__name__)

diff --git a/tests/models/test_models_unet_2d_flax.py → .../models/unets/test_models_unet_2d_flax.py b/tests/models/test_models_unet_2d_flax.py → .../models/unets/test_models_unet_2d_flax.py
diff --git a/...s/models/test_models_unet_3d_condition.py → ...ls/unets/test_models_unet_3d_condition.py b/...s/models/test_models_unet_3d_condition.py → ...ls/unets/test_models_unet_3d_condition.py
@@ -23,7 +23,7 @@
 from diffusers.utils.import_utils import is_xformers_available
 from diffusers.utils.testing_utils import enable_full_determinism, floats_tensor, skip_mps, torch_device
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 enable_full_determinism()

diff --git a/tests/models/test_models_unet_motion.py → ...s/models/unets/test_models_unet_motion.py b/tests/models/test_models_unet_motion.py → ...s/models/unets/test_models_unet_motion.py
@@ -30,7 +30,7 @@
     torch_device,
 )
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 logger = logging.get_logger(__name__)

diff --git a/...models/test_models_unet_spatiotemporal.py → .../unets/test_models_unet_spatiotemporal.py b/...models/test_models_unet_spatiotemporal.py → .../unets/test_models_unet_spatiotemporal.py
@@ -28,7 +28,7 @@
     torch_device,
 )
 
-from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
+from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 
 logger = logging.get_logger(__name__)

diff --git a/tests/models/test_unet_2d_blocks.py → tests/models/unets/test_unet_2d_blocks.py b/tests/models/test_unet_2d_blocks.py → tests/models/unets/test_unet_2d_blocks.py
diff --git a/tests/models/test_unet_blocks_common.py → ...s/models/unets/test_unet_blocks_common.py b/tests/models/test_unet_blocks_common.py → ...s/models/unets/test_unet_blocks_common.py
diff --git a/tests/pipelines/test_pipelines_common.py b/tests/pipelines/test_pipelines_common.py
@@ -38,7 +38,7 @@
     torch_device,
 )
 
-from ..models.test_models_vae import (
+from ..models.autoencoders.test_models_vae import (
     get_asym_autoencoder_kl_config,
     get_autoencoder_kl_config,
     get_autoencoder_tiny_config,