





import os
import sys
import unittest


UpperCamelCase__           =			os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
sys.path.append(os.path.join(git_repo_path, "utils"))

import check_dummies  # noqa: E402
from check_dummies import create_dummy_files, create_dummy_object, find_backend, read_init  # noqa: E402


# Align TRANSFORMERS_PATH in check_dummies with the current path
UpperCamelCase__           =			os.path.join(git_repo_path, "src", "diffusers")

class 							__SCREAMING_SNAKE_CASE       (  unittest.TestCase							):






							def 					_lowerCamelCase		(    self	):
														UpperCamelCase__					  =      find_backend("""    if not is_torch_available():"""	)
														self.assertEqual(__lowerCAmelCase	,			"""torch"""	)

														# backend_with_underscore = find_backend("    if not is_tensorflow_text_available():")
														# self.assertEqual(backend_with_underscore, "tensorflow_text")

														UpperCamelCase__					  =      find_backend("""    if not (is_torch_available() and is_transformers_available()):"""	)
														self.assertEqual(__lowerCAmelCase	,			"""torch_and_transformers"""	)

														# double_backend_with_underscore = find_backend(
														#    "    if not (is_sentencepiece_available() and is_tensorflow_text_available()):"
														# )
														# self.assertEqual(double_backend_with_underscore, "sentencepiece_and_tensorflow_text")

														UpperCamelCase__					  =      find_backend(
														    """    if not (is_torch_available() and is_transformers_available() and is_onnx_available()):"""	)
														self.assertEqual(__lowerCAmelCase	,			"""torch_and_transformers_and_onnx"""	)






							def 					_lowerCamelCase		(    self	):
														UpperCamelCase__					  =      read_init()
														# We don't assert on the exact list of keys to allow for smooth grow of backend-specific objects
														self.assertIn("""torch"""	,			__lowerCAmelCase	)
														self.assertIn("""torch_and_transformers"""	,			__lowerCAmelCase	)
														self.assertIn("""flax_and_transformers"""	,			__lowerCAmelCase	)
														self.assertIn("""torch_and_transformers_and_onnx"""	,			__lowerCAmelCase	)

														# Likewise, we can't assert on the exact content of a key
														self.assertIn("""UNet2DModel"""	,			objects["""torch"""]	)
														self.assertIn("""FlaxUNet2DConditionModel"""	,			objects["""flax"""]	)
														self.assertIn("""StableDiffusionPipeline"""	,			objects["""torch_and_transformers"""]	)
														self.assertIn("""FlaxStableDiffusionPipeline"""	,			objects["""flax_and_transformers"""]	)
														self.assertIn("""LMSDiscreteScheduler"""	,			objects["""torch_and_scipy"""]	)
														self.assertIn("""OnnxStableDiffusionPipeline"""	,			objects["""torch_and_transformers_and_onnx"""]	)






							def 					_lowerCamelCase		(    self	):
														UpperCamelCase__					  =      create_dummy_object("""CONSTANT"""	,			"""'torch'"""	)
														self.assertEqual(__lowerCAmelCase	,			"""\nCONSTANT = None\n"""	)

														UpperCamelCase__					  =      create_dummy_object("""function"""	,			"""'torch'"""	)
														self.assertEqual(
														    __lowerCAmelCase	,			"""\ndef function(*args, **kwargs):\n    requires_backends(function, 'torch')\n"""	)

														UpperCamelCase__					  =      """
class FakeClass(metaclass=DummyObject):
    _backends = 'torch'

    def __init__(self, *args, **kwargs):
        requires_backends(self, 'torch')

    @classmethod
    def from_config(cls, *args, **kwargs):
        requires_backends(cls, 'torch')

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, 'torch')
"""
														UpperCamelCase__					  =      create_dummy_object("""FakeClass"""	,			"""'torch'"""	)
														self.assertEqual(__lowerCAmelCase	,			__lowerCAmelCase	)







							def 					_lowerCamelCase		(    self	):
														UpperCamelCase__					  =      """# This file is autogenerated by the command `make fix-copies`, do not edit.
from ..utils import DummyObject, requires_backends


CONSTANT = None


def function(*args, **kwargs):
    requires_backends(function, [\"torch\"])


class FakeClass(metaclass=DummyObject):
    _backends = [\"torch\"]

    def __init__(self, *args, **kwargs):
        requires_backends(self, [\"torch\"])

    @classmethod
    def from_config(cls, *args, **kwargs):
        requires_backends(cls, [\"torch\"])

    @classmethod
    def from_pretrained(cls, *args, **kwargs):
        requires_backends(cls, [\"torch\"])
"""
														UpperCamelCase__					  =      create_dummy_files({"""torch""": ["""CONSTANT""", """function""", """FakeClass"""]}	)
														self.assertEqual(dummy_files["""torch"""]	,			__lowerCAmelCase	)











# this script reports modified .py files under the desired list of top-level sub-dirs passed as a list of arguments, e.g.:
#   python ./utils/get_modified_files.py utils src tests examples
#
# it uses git to find the forking point and which files were modified - i.e. files not under git won't be considered
# since the output of this script is fed into Makefile commands it doesn't print a newline after the results

import re
import subprocess
import sys


UpperCamelCase__           =			subprocess.check_output("git merge-base main HEAD".split()).decode("utf-8")
UpperCamelCase__           =			subprocess.check_output(f"""git diff --name-only {fork_point_sha}""".split()).decode("utf-8").split()

UpperCamelCase__           =			"|".join(sys.argv[1:])
UpperCamelCase__           =			re.compile(rf"""^({joined_dirs}).*?\.py$""")

UpperCamelCase__           =			[x for x in modified_files if regex.match(x)]
print(" ".join(relevant_modified_files), end="")




