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