BVh~dZddlZddlmZddlmZddlmZddlmZgdZ ddZ d Z Gd d ejZ Gd d ejZejj dZej$ejee efzZej&Zej(Zej*Zy)zHThis module customizes `test_combinations` for `tf.keras` related tests.N)tf2) combinations)test_combinations) testing_utils) functionalsubclass sequentialc|tjrdgnddg}|ddg}g}d|vr|tjdg|z }d|vr|tjdgdgz }|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. eagergraphTF)mode run_eagerly)renabledrcombine)r rresults T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/keras/combinations.pykeras_mode_combinationsrsz" \ G9GW+=D-K & _ l"" {KKF _ l"" wGGF -c6tjtS)N) model_type)rrKERAS_MODEL_TYPESrrkeras_model_type_combinationsr8s   ): ;;rceZdZdZdZdZy)KerasModeCombinationzfCombination for Keras test mode. It by default includes v1_session, v2_eager and v2_tf_function. cZ|jdd}|tj|gSgSNr)poprrun_eagerly_scope)selfkwargsrs rcontext_managersz%KerasModeCombination.context_managersBs1**]D1K--k: ;; irc.tjdgSrrOptionalParameterr s rparameter_modifiersz(KerasModeCombination.parameter_modifiersJs  / / > ??rN__name__ __module__ __qualname____doc__r"r'rrrrr<s @rrceZdZdZdZdZy)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. cf|jdd}|tvrtj|gSgSNr)rrrmodel_type_scope)r r!rs rr"z*KerasModelTypeCombination.context_managersXs4L$/J&&,,Z8 99 irc.tjdgSr0r$r&s rr'z-KerasModelTypeCombination.parameter_modifiers_s  / / = >>rNr(rrrr.r.Ns?rr.r)NN)r, functoolstensorflow.pythonrtensorflow.python.frameworkrrtensorflow.python.kerasrrrrTestCombinationrr.generatekeywords _defaultspartialrtimes NamedObjectrrrr>sO!491<:<@,<<@$? 1 A A?*  ! ! * *+> ? 9  689: ;  # #++ r