From 255fa32aab43bc068e51814794a905ed19e3cc81 Mon Sep 17 00:00:00 2001 From: lmanan Date: Wed, 11 Oct 2023 21:34:26 -0400 Subject: [PATCH] Specify default = None for validate_data_config --- cellulus/configs/train_config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cellulus/configs/train_config.py b/cellulus/configs/train_config.py index f6e514e..c38a9f6 100644 --- a/cellulus/configs/train_config.py +++ b/cellulus/configs/train_config.py @@ -77,7 +77,7 @@ class TrainConfig: Configuration object for the training data. - validate_data_config: + validate_data_config (default = None): Configuration object for the validation data. @@ -91,7 +91,7 @@ class TrainConfig: train_data_config: DatasetConfig = attrs.field(converter=to_config(DatasetConfig)) validate_data_config: DatasetConfig = attrs.field( - converter=to_config(DatasetConfig) + default=None, converter=to_config(DatasetConfig) ) crop_size: List = attrs.field(default=[252, 252], validator=instance_of(List)) batch_size: int = attrs.field(default=8, validator=instance_of(int))