From 48feca832c77239909bc6dd8e265e8f3513c65ae Mon Sep 17 00:00:00 2001 From: davebulaval Date: Mon, 1 Jul 2024 12:55:18 -0400 Subject: [PATCH] fix model filename --- CHANGELOG.md | 4 ++++ .../contract_faker/tvae_synthetic_fpq1_faker.py | 2 +- .../{tvae_model_q1_2023.pkl => tvae_model_v1.pkl} | Bin 3 files changed, 5 insertions(+), 1 deletion(-) rename risc_generator/resources/{tvae_model_q1_2023.pkl => tvae_model_v1.pkl} (100%) diff --git a/CHANGELOG.md b/CHANGELOG.md index 718260b..d3db7d4 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -3,3 +3,7 @@ The initial functionalities of simulation for automobile insurance contracts for # 0.0.2 - Fixed error in a title of the generated text. + +# 0.0.3 +- Fix model filename and loading path. + diff --git a/risc_generator/faker/contract_faker/tvae_synthetic_fpq1_faker.py b/risc_generator/faker/contract_faker/tvae_synthetic_fpq1_faker.py index 5921815..9ddb12e 100644 --- a/risc_generator/faker/contract_faker/tvae_synthetic_fpq1_faker.py +++ b/risc_generator/faker/contract_faker/tvae_synthetic_fpq1_faker.py @@ -24,7 +24,7 @@ def __init__( """ if tvae_synthetic_data_faker_model_path is None: tvae_synthetic_data_faker_model_path = ( - os.path.join(os.path.dirname(__file__), '../../resources', "tvae_model_q1_2023.pkl"), + os.path.join(os.path.dirname(__file__), '../../resources', "tvae_model_v1.pkl"), ) protection_faker = ProtectionsFaker( model=TVAESyntheticProtectionsModelWrapper(tvae_synthetic_data_faker_model_path) diff --git a/risc_generator/resources/tvae_model_q1_2023.pkl b/risc_generator/resources/tvae_model_v1.pkl similarity index 100% rename from risc_generator/resources/tvae_model_q1_2023.pkl rename to risc_generator/resources/tvae_model_v1.pkl