


import math







class      lowerCAmelCase__       :




     '''simple docstring'''


     def __init__(						self ,						__lowerCamelCase=0)			->				str:  # a graph with Node 0,1,...,N-1
            _A      :       Tuple				    = n
            _A      :       Optional[int]				    = [
                [math.inf for j in range(0 ,						__lowerCamelCase)] for i in range(0 ,						__lowerCamelCase)
            ]  # adjacency matrix for weight
            _A      :       List[str]				    = [
                [math.inf for j in range(0 ,						__lowerCamelCase)] for i in range(0 ,						__lowerCamelCase)
            ]  # dp[i][j] stores minimum distance from i to j


     def  _lowerCamelCase (						self ,						__lowerCamelCase ,						__lowerCamelCase ,						__lowerCamelCase)			->				int:
            _A      :       Dict				    = w


     def  _lowerCamelCase (						self)			->				Union[str, Any]:
            for k in range(0 ,						self.n):
                   for i in range(0 ,						self.n):
                          for j in range(0 ,						self.n):
                                 _A      :       Optional[Any]				    = min(self.dp[i][j] ,						self.dp[i][k] + self.dp[k][j])


     def  _lowerCamelCase (						self ,						__lowerCamelCase ,						__lowerCamelCase)			->				Optional[int]:
            return self.dp[u][v]


if __name__ == "__main__":
    lowerCAmelCase__      		= Graph(5)
    graph.add_edge(0, 2, 9)
    graph.add_edge(0, 4, 10)
    graph.add_edge(1, 3, 5)
    graph.add_edge(2, 3, 7)
    graph.add_edge(3, 0, 10)
    graph.add_edge(3, 1, 2)
    graph.add_edge(3, 2, 1)
    graph.add_edge(3, 4, 6)
    graph.add_edge(4, 1, 3)
    graph.add_edge(4, 2, 4)
    graph.add_edge(4, 3, 9)
    graph.floyd_warshall()
    graph.show_min(1, 4)
    graph.show_min(0, 3)




'''simple docstring'''
import unittest
from dataclasses import dataclass

import pytest

from accelerate.commands.config.config_args import SageMakerConfig
from accelerate.utils import ComputeEnvironment
from accelerate.utils.launch import _convert_nargs_to_dict
@dataclass
class    lowercase	(							A__					):

		"""simple docstring"""

		_a     =				ComputeEnvironment.AMAZON_SAGEMAKER
		_a     =				True
		_a     =				'ml.p3.2xlarge'
		_a     =				'accelerate_sagemaker_execution_role'
		_a     =				'hf-sm'
		_a     =				'us-east-1'
		_a     =				1
		_a     =				'accelerate-sagemaker-1'
		_a     =				'1.6'
		_a     =				'4.4'
		_a     =				'train.py'
		_a     =				[
		    '--model_name_or_path',
		    'bert',
		    '--do_train',
		    'False',
		    '--epochs',
		    '3',
		    '--learning_rate',
		    '5e-5',
		    '--max_steps',
		    '50.5',
		]
		_a     =				[
		    '--model_name_or_path',
		    'bert',
		    '--do_train',
		    '--do_test',
		    'False',
		    '--do_predict',
		    '--epochs',
		    '3',
		    '--learning_rate',
		    '5e-5',
		    '--max_steps',
		    '50.5',
		]



class    lowercase	(							unittest.TestCase					):

		"""simple docstring"""

		def     lowerCAmelCase__				(		self							):

							'''simple docstring'''


							UpperCamelCase__   :Union[str, Any]												= _convert_nargs_to_dict(MockLaunchConfig.success_training_script_args							)
							assert isinstance(converted_args['''model_name_or_path''']    ,   UpperCamelCase_							)
							assert isinstance(converted_args['''do_train''']    ,   UpperCamelCase_							)
							assert isinstance(converted_args['''epochs''']    ,   UpperCamelCase_							)
							assert isinstance(converted_args['''learning_rate''']    ,   UpperCamelCase_							)
							assert isinstance(converted_args['''max_steps''']    ,   UpperCamelCase_							)

							with pytest.raises(UpperCamelCase_							):
												_convert_nargs_to_dict(MockLaunchConfig.fail_training_script_args							)