o 2h@sdZddlZddlmZddlmZddlmZddlmZgdZ ddd Z d d Z Gd d d ej Z Gdddej ZejjdZejejee efdZejZejZejZdS)zHThis module customizes `test_combinations` for `tf.keras` related tests.N)tf2) combinations)test_combinations) testing_utils) functionalsubclass sequentialcCsp|durtr dgnddg}|durddg}g}d|vr'|tjdg|d7}d|vr6|tjdgdgd7}|S)aReturns the default test combinations for tf.keras tests. Note that if tf2 is enabled, then v1 session test will be skipped. Args: mode: List of modes to run the tests. The valid options are 'graph' and 'eager'. Default to ['graph', 'eager'] if not specified. If a empty list is provide, then the test will run under the context based on tf's version, eg graph for v1 and eager for v2. run_eagerly: List of `run_eagerly` value to be run with the tests. Default to [True, False] if not specified. Note that for `graph` mode, run_eagerly value will only be False. Returns: A list contains all the combinations to be used to generate test cases. NeagergraphTF)mode run_eagerly)renabledrcombine)r r resultr_/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/keras/combinations.pykeras_mode_combinationssrcCs tjtdS)N) model_type)rrKERAS_MODEL_TYPESrrrrkeras_model_type_combinations8 rc@ eZdZdZddZddZdS)KerasModeCombinationzfCombination for Keras test mode. It by default includes v1_session, v2_eager and v2_tf_function. cCs$|dd}|durt|gSgSNr )poprrun_eagerly_scope)selfkwargsr rrrcontext_managersBs  z%KerasModeCombination.context_managerscC tdgSrrOptionalParameterrrrrparameter_modifiersJrz(KerasModeCombination.parameter_modifiersN__name__ __module__ __qualname____doc__rr#rrrrr<s rc@r)KerasModelTypeCombinationaKCombination for Keras model types when doing model test. It by default includes 'functional', 'subclass', 'sequential'. Various methods in `testing_utils` to get models will auto-generate a model of the currently active Keras model type. This allows unittests to confirm the equivalence between different Keras models. cCs$|dd}|tvrt|gSgSNr)rrrmodel_type_scope)rrrrrrrXs  z*KerasModelTypeCombination.context_managerscCrr*r r"rrrr#_rz-KerasModelTypeCombination.parameter_modifiersNr$rrrrr)Ns r)r)NN)r( functoolstensorflow.pythonrtensorflow.python.frameworkrrtensorflow.python.kerasrrrrTestCombinationrr)generatekeywords _defaultspartialrtimes NamedObjectrrrrs*