diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py index e774610..b177e95 100644 --- a/aimodel/src/lib/ai/RainfallWaterContraster.py +++ b/aimodel/src/lib/ai/RainfallWaterContraster.py @@ -53,7 +53,7 @@ class RainfallWaterContraster(object): @staticmethod def from_checkpoint(filepath_checkpoint, filepath_hyperparams=None): if not filepath_checkpoint: - filepath_checkpoint = find_paramsjson(filepath_checkpoint) + filepath_hyperparams = find_paramsjson(filepath_checkpoint) hyperparams = json.loads(readfile(filepath_hyperparams)) return RainfallWaterContraster(filepath_checkpoint=filepath_checkpoint, **hyperparams)