a \=ic@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)Z functionalsubclass sequentialcCsp|durtrdgnddg}|dur.ddg}g}d|vrN|tjdg|d7}d|vrl|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 resultru/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorflow/python/keras/combinations.pykeras_mode_combinationssrcCs tjtdS)N) model_type)rr KERAS_MODEL_TYPESrrrrkeras_model_type_combinations8src@s eZdZdZddZddZdS)KerasModeCombinationzfCombination for Keras test mode. It by default includes v1_session, v2_eager and v2_tf_function. cCs(|dd}|dur t|gSgSdSNr )poprZrun_eagerly_scope)selfkwargsr rrrcontext_managersBs  z%KerasModeCombination.context_managerscCs tdgSrrZOptionalParameterrrrrparameter_modifiersJsz(KerasModeCombination.parameter_modifiersN__name__ __module__ __qualname____doc__rrrrrrr<src@s eZdZdZddZddZdS)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}|tvr t|gSgSdSNr)rrrZmodel_type_scope)rrrrrrrXs  z*KerasModelTypeCombination.context_managerscCs tdgSr$rrrrrr_sz-KerasModelTypeCombination.parameter_modifiersNrrrrrr#Ns r#r)NN)r" functoolsZtensorflow.pythonrZtensorflow.python.frameworkrrZtensorflow.python.kerasrrrrZTestCombinationrr#generatekeywords _defaultspartialr timesZ NamedObjectrrrrs(