Skip to content

Commit 0f55ba2

Browse files
authored
Merge pull request #272 from theislab/fix/predict_kwargs
fix predict kwargs
2 parents ee9c3fa + 9b1e3bd commit 0f55ba2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/cellflow/model/_cellflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ def prepare_validation_data(
237237
split_val = len(val_data.control_to_perturbation) > 1
238238
predict_kwargs = predict_kwargs or {}
239239
# Check if predict_kwargs is alreday provided from an earlier call
240-
if "predict_kwargs" in self._validation_data:
240+
if "predict_kwargs" in self._validation_data and len(predict_kwargs):
241241
predict_kwargs = self._validation_data["predict_kwargs"].update(predict_kwargs)
242242
# Set batched prediction to False if split_val is True
243243
if split_val:

0 commit comments

Comments
 (0)