





"""simple docstring"""
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier

# Load iris file
__UpperCamelCase    :	Tuple						=     load_iris()
iris.keys()


print(f'''Target names: \n {iris.target_names} ''')
print(f'''\n Features: \n {iris.feature_names}''')

# Train set e Test set
__UpperCamelCase					,__UpperCamelCase					,__UpperCamelCase					,__UpperCamelCase    :	Tuple						=     train_test_split(
    iris['''data'''], iris['''target'''], random_state=4
)

# KNN

__UpperCamelCase    :	Optional[int]						=     KNeighborsClassifier(n_neighbors=1)
knn.fit(X_train, y_train)

# new array to test
__UpperCamelCase    :	Optional[int]						=     [[1, 2, 1, 4], [2, 3, 4, 5]]

__UpperCamelCase    :	Union[str, Any]						=     knn.predict(X_new)

print(
    f'''\nNew array: \n {X_new}\n\nTarget Names Prediction: \n'''
    f''' {iris['target_names'][prediction]}'''
)












"""simple docstring"""
import unittest
from pathlib import Path

from transformers import is_vision_available, load_tool
from transformers.testing_utils import get_tests_dir

from .test_tools_common import ToolTesterMixin


if is_vision_available():
       from PIL import Image







class 			a       (      unittest.TestCase      ,   a__     ):



 def       UpperCamelCase__     (    self    ):







  """simple docstring"""






  lowerCAmelCase													=   load_tool('image-question-answering'    )
  self.tool.setup()
  lowerCAmelCase													=   load_tool('image-question-answering'							,					remote=_snake_case    )



 def       UpperCamelCase__     (    self    ):







  """simple docstring"""






  lowerCAmelCase													=   Image.open(Path(get_tests_dir('fixtures/tests_samples/COCO'    )    ) / '000000039769.png'    )
  lowerCAmelCase													=   self.tool(_snake_case							,					'How many cats are sleeping on the couch?'    )
  self.assertEqual(_snake_case							,					'2'    )



 def       UpperCamelCase__     (    self    ):







  """simple docstring"""






  lowerCAmelCase													=   Image.open(Path(get_tests_dir('fixtures/tests_samples/COCO'    )    ) / '000000039769.png'    )
  lowerCAmelCase													=   self.remote_tool(_snake_case							,					'How many cats are sleeping on the couch?'    )
  self.assertEqual(_snake_case							,					'2'    )



 def       UpperCamelCase__     (    self    ):







  """simple docstring"""






  lowerCAmelCase													=   Image.open(Path(get_tests_dir('fixtures/tests_samples/COCO'    )    ) / '000000039769.png'    )
  lowerCAmelCase													=   self.tool(image=_snake_case							,					question='How many cats are sleeping on the couch?'    )
  self.assertEqual(_snake_case							,					'2'    )



 def       UpperCamelCase__     (    self    ):







  """simple docstring"""






  lowerCAmelCase													=   Image.open(Path(get_tests_dir('fixtures/tests_samples/COCO'    )    ) / '000000039769.png'    )
  lowerCAmelCase													=   self.remote_tool(image=_snake_case							,					question='How many cats are sleeping on the couch?'    )
  self.assertEqual(_snake_case							,					'2'    )





