From 105dc5bc5645f1c2e4ae30df759190550fbdd5bf Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 24 Nov 2022 17:51:29 +0000 Subject: [PATCH] missing kwargs --- aimodel/src/lib/dataset/dataset_mono.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset_mono.py b/aimodel/src/lib/dataset/dataset_mono.py index 591208a..1a397e9 100644 --- a/aimodel/src/lib/dataset/dataset_mono.py +++ b/aimodel/src/lib/dataset/dataset_mono.py @@ -109,7 +109,7 @@ def dataset_mono(dirpath_input, train_percentage=0.8, **kwargs): return dataset_train, dataset_validate #, filepaths -def dataset_mono_predict(dirpath_input): +def dataset_mono_predict(dirpath_input, **kwargs): """Creates a tf.data.Dataset() for prediction using the contrastive learning model. Note that this WILL MANGLE THE ORDERING if you set parallel_reads_multiplier to anything other than 0!!