fix: fix path parsing and default in hybrid serving

zama-ai · Sep 22, 2023 · afd049a · afd049a
1 parent dd16433
commit afd049a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/use_case_examples/hybrid_model/serve_model.py b/use_case_examples/hybrid_model/serve_model.py
@@ -39,13 +39,13 @@ def underscore_str_to_tuple(tup):
         "--path-to-models",
         dest="path_to_models",
         type=Path,
-        default=FILE_FOLDER / Path("user_keys"),
+        default=FILE_FOLDER / Path("compiled_models"),
     )
     parser.add_argument(
         "--path-to-keys",
         dest="path_to_keys",
         type=Path,
-        default=FILE_FOLDER / Path("compiled_models"),
+        default=FILE_FOLDER / Path("user_keys"),
     )
     args = parser.parse_args()
     app = FastAPI(debug=False)
@@ -61,6 +61,8 @@ def underscore_str_to_tuple(tup):
     # We build the following mapping:
     # model_name -> module_name -> input_shape -> some information
     for model_path in MODELS_PATH.iterdir():  # Model
+        if not model_path.is_dir():
+            continue
         model_name = model_path.name
         MODULES[model_name] = defaultdict(dict)
         for module_path in model_path.iterdir():  # Module