From 606be6e74801039ec0c4012c7745e5241238e4d0 Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Wed, 24 Jul 2024 07:56:59 -0700 Subject: [PATCH] Add missing list of models Signed-off-by: Mihai Maruseac --- model_signing/test_support.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/model_signing/test_support.py b/model_signing/test_support.py index 6d8aefaf..ab28fc67 100644 --- a/model_signing/test_support.py +++ b/model_signing/test_support.py @@ -42,6 +42,15 @@ ] +# All directory models to use in testing, where only non empty directory models +# are supported. See also `all_test_models` comments. +all_non_empty_directory_test_models = [ + "sample_model_folder", + "deep_model_folder", + "model_folder_with_empty_file", +] + + def get_first_directory(path: pathlib.Path) -> pathlib.Path: """Returns the first directory that is a children of path.