



# tests directory-specific settings - this file is run automatically
# by pytest before any tests are run

import sys
import warnings
from os.path import abspath, dirname, join


# allow having multiple repository checkouts and not needing to remember to rerun
# 'pip install -e .[dev]' when switching between checkouts and running tests.
A__								=						abspath(join(dirname(dirname(dirname(__file__))), '''src'''))
sys.path.insert(1, git_repo_path)


# silence FutureWarning warnings in tests since often we can't act on them until
# they become normal warnings - i.e. the tests still need to test the current functionality
warnings.simplefilter(action='''ignore''', category=FutureWarning)



def 			_lowerCAmelCase       (					__lowerCAmelCase      )   ->					str:


				"""simple docstring"""

				from transformers.testing_utils import pytest_addoption_shared

				pytest_addoption_shared(__lowerCAmelCase      )





def 			_lowerCAmelCase       (					__lowerCAmelCase      )   ->					List[Any]:


				"""simple docstring"""

				from transformers.testing_utils import pytest_terminal_summary_main

				snake_case__	:      Dict     		=	terminalreporter.config.getoption('''--make-reports'''      )
				if make_reports:
								pytest_terminal_summary_main(__lowerCAmelCase  ,  id=__lowerCAmelCase      )






from torch import nn


class       a							(			nn.Module				):



		def __init__(     self     :Tuple   ,__lowercase     :Optional[int]   ,__lowercase     :int  ):
						super().__init__()
						snake_case__	:      Optional[Any]     		=	class_size
						snake_case__	:      Dict     		=	embed_size
						# self.mlp1 = nn.Linear(embed_size, embed_size)
						# self.mlp2 = (nn.Linear(embed_size, class_size))
						snake_case__	:      Dict     		=	nn.Linear(__lowercase   ,__lowercase  )






		def 	__lowerCamelCase			(     self     :str   ,__lowercase     :int  ):
						# hidden_state = nn.functional.relu(self.mlp1(hidden_state))
						# hidden_state = self.mlp2(hidden_state)
						snake_case__	:      Optional[Any]     		=	self.mlp(__lowercase  )
						return logits

