diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 11944cc..904e271 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -29,12 +29,22 @@ from numpy import ndarray from typing import Any import gc from sientia_do.temporal.constants import DATETIME_FORMAT_WITH_TZ +import ctypes ARTIFACTS_PATH = "./tmp/artifacts" TRANSFORMED_COMPRESSED_PATH = "artifacts/training_transformer.pkl" PREDICTION_COMPRESSED_PATH = "artifacts/stacking_model.pkl" +def force_memory_release(): + gc.collect() + + try: + ctypes.CDLL("libc.so.6").malloc_trim(0) + except: + pass + + class MLFlowRepository(): def __init__(self, host: str, username: str, password: str, logger: Logger):