



import pytest

import datasets.config
from datasets.utils.info_utils import is_small_dataset

@pytest.mark.parametrize("dataset_size"    ,   [None, 400 * 2**20, 600 * 2**20]    )
@pytest.mark.parametrize("input_in_memory_max_size"    ,   ["default", 0, 100 * 2**20, 900 * 2**20]    )
def    lowerCamelCase_(lowerCamelCase_    ,   lowerCamelCase_    ,   lowerCamelCase_    ) ->       Tuple:
						if input_in_memory_max_size != "default":
												monkeypatch.setattr(datasets.config    ,   "IN_MEMORY_MAX_SIZE"    ,   lowerCamelCase_    )
						UpperCAmelCase         =     datasets.config.IN_MEMORY_MAX_SIZE
						if input_in_memory_max_size == "default":
												assert in_memory_max_size == 0
						else:
												assert in_memory_max_size == input_in_memory_max_size
						if dataset_size and in_memory_max_size:
												UpperCAmelCase         =     dataset_size < in_memory_max_size
						else:
												UpperCAmelCase         =     False
						UpperCAmelCase         =     is_small_dataset(lowerCamelCase_    )
						assert result == expected








from argparse import ArgumentParser

from .env import EnvironmentCommand

def    lowerCamelCase_() ->       Any:
						UpperCAmelCase         =     ArgumentParser("Diffusers CLI tool"    ,   usage="diffusers-cli <command> [<args>]"    )
						UpperCAmelCase         =     parser.add_subparsers(help="diffusers-cli command helpers"    )

						# Register commands
						EnvironmentCommand.register_subcommand(lowerCamelCase_    )

						# Let's go
						UpperCAmelCase         =     parser.parse_args()

						if not hasattr(lowerCamelCase_    ,   "func"    ):
												parser.print_help()
												exit(1    )

						# Run
						UpperCAmelCase         =     args.func(lowerCamelCase_    )
						service.run()


if __name__ == "__main__":
		main()



