From 931db5860ea1f1f0bb505842f7387d47d7569e55 Mon Sep 17 00:00:00 2001 From: JulienPeloton Date: Fri, 15 Sep 2023 08:09:04 +0200 Subject: [PATCH] Add new models for deployment --- setup.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 827ae5df..ded5bf6d 100644 --- a/setup.py +++ b/setup.py @@ -41,12 +41,15 @@ 'data/models/snn_models/*/*.pt', 'data/models/snn_models/*/*.json', 'data/models/snn_models/*/*.txt', - 'data/models/cats_models/model_meta_0/*.pb', - 'data/models/cats_models/model_meta_0/variables/variables*', + 'data/models/snn_models/elasticc_binary_broad/*/*.pt', + 'data/models/snn_models/elasticc_binary_broad/*/*.json', + 'data/models/snn_models/elasticc_binary_broad/*/*.txt', + 'data/models/cats_models/model_fold0.h5', 'data/models/t2/*/*.pb', 'data/models/t2/*/*/variables*', 'data/models/anomaly_detection/*.csv', 'data/models/anomaly_detection/*.zip', + 'data/models/for_al_loop/model_20230907.pkl', 'data/catalogs/*.parquet', 'data/catalogs/*.fits', 'data/catalogs/*.csv'],