

"""simple docstring"""




import argparse
import os


__A     :    Optional[int]    =		{
    'pytorch': {
        '1.13': {
            'torch': '1.13.1',
            'torchvision': '0.14.1',
            'torchaudio': '0.13.1',
            'python': 3.9,
            'cuda': 'cu116',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1'
                ' --extra-index-url https://download.pytorch.org/whl/cu116'
            ),
            'base_image': 'nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04',
        },
        '1.12': {
            'torch': '1.12.1',
            'torchvision': '0.13.1',
            'torchaudio': '0.12.1',
            'python': 3.9,
            'cuda': 'cu113',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1'
                ' --extra-index-url https://download.pytorch.org/whl/cu113'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '1.11': {
            'torch': '1.11.0',
            'torchvision': '0.12.0',
            'torchaudio': '0.11.0',
            'python': 3.9,
            'cuda': 'cu113',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0'
                ' --extra-index-url https://download.pytorch.org/whl/cu113'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '1.10': {
            'torch': '1.10.2',
            'torchvision': '0.11.3',
            'torchaudio': '0.10.2',
            'python': 3.9,
            'cuda': 'cu113',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.10.2 torchvision==0.11.3 torchaudio==0.10.2'
                ' --extra-index-url https://download.pytorch.org/whl/cu113'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        # torchaudio < 0.10 has no CUDA-enabled binary distributions
        '1.9': {
            'torch': '1.9.1',
            'torchvision': '0.10.1',
            'torchaudio': '0.9.1',
            'python': 3.9,
            'cuda': 'cu111',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1'
                ' --extra-index-url https://download.pytorch.org/whl/cu111'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
    },
    'tensorflow': {
        '2.11': {
            'tensorflow': '2.11.1',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.11.1',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.10': {
            'tensorflow': '2.10.1',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.10.1',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.9': {
            'tensorflow': '2.9.3',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.9.3',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.8': {
            'tensorflow': '2.8.2',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.8.2',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.7': {
            'tensorflow': '2.7.3',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.7.3',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.6': {
            'tensorflow': '2.6.5',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.6.5',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.5': {
            'tensorflow': '2.5.3',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.5.3',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
    },
}


if __name__ == "__main__":
	__A     :    Optional[Any]    =		argparse.ArgumentParser('Choose the framework and version to install')
	parser.add_argument(
	    '--framework', help='The framework to install. Should be `torch` or `tensorflow`', type=str, required=True
	)
	parser.add_argument('--version', help='The version of the framework to install.', type=str, required=True)
	__A     :    Optional[Any]    =		parser.parse_args()

	__A     :    str    =		past_versions_testing[args.framework][args.version]

	os.system(f"""echo \"export INSTALL_CMD=\'{info['install']}\'\" >> ~/.profile""")
	print(f"""echo \"export INSTALL_CMD=\'{info['install']}\'\" >> ~/.profile""")

	__A     :    Union[str, Any]    =		''
	if args.framework == "pytorch":
		__A     :    Union[str, Any]    =		info['cuda']
	os.system(f"""echo \"export CUDA=\'{cuda}\'\" >> ~/.profile""")
	print(f"""echo \"export CUDA=\'{cuda}\'\" >> ~/.profile""")












from __future__ import annotations

from collections import deque
from queue import Queue
from timeit import timeit

_lowerCamelCase       =	{
    'A': ['B', 'C'],
    'B': ['A', 'D', 'E'],
    'C': ['A', 'F'],
    'D': ['B'],
    'E': ['B', 'F'],
    'F': ['C', 'E'],
}






def  SCREAMING_SNAKE_CASE__							(				UpperCamelCase__:							dict    ,      UpperCamelCase__:							str			):
		SCREAMING_SNAKE_CASE__								=					{start}
		SCREAMING_SNAKE_CASE__								=					[start]
		SCREAMING_SNAKE_CASE__								=					Queue()
		queue.put(UpperCamelCase__			)
		while not queue.empty():
				SCREAMING_SNAKE_CASE__								=					queue.get()
				for w in graph[v]:
						if w not in explored:
								explored.add(UpperCamelCase__			)
								result.append(UpperCamelCase__			)
								queue.put(UpperCamelCase__			)
		return result






def  SCREAMING_SNAKE_CASE__							(				UpperCamelCase__:							dict    ,      UpperCamelCase__:							str			):
		SCREAMING_SNAKE_CASE__								=					{start}
		SCREAMING_SNAKE_CASE__								=					[start]
		SCREAMING_SNAKE_CASE__								=					deque([start]			)
		while queue:
				SCREAMING_SNAKE_CASE__								=					queue.popleft()
				for child in graph[v]:
						if child not in visited:
								visited.add(UpperCamelCase__			)
								result.append(UpperCamelCase__			)
								queue.append(UpperCamelCase__			)
		return result






def  SCREAMING_SNAKE_CASE__							(				UpperCamelCase__:							str			):
		SCREAMING_SNAKE_CASE__								=					f'''from __main__ import G, {name}'''
		SCREAMING_SNAKE_CASE__								=					10_000
		SCREAMING_SNAKE_CASE__								=					timeit(f'''{name}(G, \'A\')'''    ,      setup=UpperCamelCase__    ,      number=UpperCamelCase__			)
		print(f'''{name:<35} finished {number} runs in {res:.5f} seconds'''			)


if __name__ == "__main__":
	import doctest

	doctest.testmod()

	benchmark_function('breadth_first_search')
	benchmark_function('breadth_first_search_with_deque')
	# breadth_first_search                finished 10000 runs in 0.20999 seconds
	# breadth_first_search_with_deque     finished 10000 runs in 0.01421 seconds