


"""simple docstring"""


import unittest
from pathlib import Path
from tempfile import TemporaryDirectory
from unittest.mock import Mock, patch

import diffusers.utils.hub_utils
class snake_case   (	unittest.TestCase):







		@patch("diffusers.utils.hub_utils.get_full_repo_name"		)
		def 			a_			(       self   :							Any     , a__   :							Mock		)							->    None:





						'''simple docstring'''







						_A       	=      "full_repo_name"
						with TemporaryDirectory() as tmpdir:
										# Dummy args values
										_A       	=      Mock()
										_A       	=      tmpdir
										_A       	=      0
										_A       	=      "hub_token"
										_A       	=      "dataset_name"
										_A       	=      0.0_1
										_A       	=      10_00_00
										_A       	=      1_00_00
										_A       	=      0.0_1
										_A       	=      0.0_2
										_A       	=      0.0_3
										_A       	=      0.0_0_0_5
										_A       	=      0.0_0_0_0_0_1
										_A       	=      1
										_A       	=      10
										_A       	=      0.0_0_1
										_A       	=      0.1
										_A       	=      0.2
										_A       	=      True

										# Model card mush be rendered and saved
										diffusers.utils.hub_utils.create_model_card(a__     , model_name="model_name"		)
										self.assertTrue((Path(a__		) / "README.md").is_file()		)



"""simple docstring"""


import os

from apache_beam.io.filesystems import FileSystems
from apache_beam.pipeline import Pipeline

from .logging import get_logger


a_											=     2 << 20  # 2mb
a_											=     get_logger(__name__)
class snake_case   (	_UpperCamelCase):







		def 			a_			(       self   :							Tuple		)							->    Optional[int]:





						'''simple docstring'''







						_A       	=      self._options.get_all_options().get("runner"		)
						return runner in [None, "DirectRunner", "PortableRunner"]






def   a__			(					__lowercase  ,      __lowercase  ,      __lowercase=False )						->						Any:
				_A       	=      FileSystems
				if fs.exists(__lowercase ):
								if force_upload:
												logger.info(f"""Remote path already exist: {remote_file_path}. Overwriting it as force_upload=True.""" )
								else:
												logger.info(f"""Remote path already exist: {remote_file_path}. Skipping it as force_upload=False.""" )
												return
				with fs.create(__lowercase ) as remote_file:
								with open(__lowercase  ,      "rb" ) as local_file:
												_A       	=      local_file.read(__lowercase )
												while chunk:
																remote_file.write(__lowercase )
																_A       	=      local_file.read(__lowercase )






def   a__			(					__lowercase  ,      __lowercase  ,      __lowercase=False )						->						Union[str, Any]:
				_A       	=      FileSystems
				if os.path.exists(__lowercase ):
								if force_download:
												logger.info(f"""Local path already exist: {remote_file_path}. Overwriting it as force_upload=True.""" )
								else:
												logger.info(f"""Local path already exist: {remote_file_path}. Skipping it as force_upload=False.""" )
												return
				with fs.open(__lowercase ) as remote_file:
								with open(__lowercase  ,      "wb" ) as local_file:
												_A       	=      remote_file.read(__lowercase )
												while chunk:
																local_file.write(__lowercase )
																_A       	=      remote_file.read(__lowercase )