Skip to content

Commit 6014e30

Browse files
authored
Apply suggestions from code review
1 parent 143650d commit 6014e30

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

art/estimators/generation/tensorflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def __init__(
208208
self,
209209
encoding_length: int,
210210
model: "tf.Tensor",
211-
channels_first=False,
211+
channels_first: bool = False,
212212
clip_values: Optional["CLIP_VALUES_TYPE"] = None,
213213
preprocessing_defences: Union["Preprocessor", List["Preprocessor"], None] = None,
214214
postprocessing_defences: Union["Postprocessor", List["Postprocessor"], None] = None,

0 commit comments

Comments
 (0)