fixed small bug in one of the examples

jrzaurin · jrzaurin · commit 9fad33313ab2 · 2023-11-01T17:48:21.000Z
diff --git a/examples/scripts/adult_census.py b/examples/scripts/adult_census.py
@@ -164,4 +164,4 @@
         )
 
         # 3. Either fit or directly predict
-        preds = trainer_new.predict(X_wide=X_wide, X_tab=X_tab)
+        preds = trainer_new.predict(X_wide=X_wide, X_tab=X_tab, batch_size=256)

Original file line number	Diff line number	Diff line change
`@@ -164,4 +164,4 @@`
`164`	`164`	`)`
`165`	`165`
`166`	`166`	`# 3. Either fit or directly predict`
`167`		`- preds = trainer_new.predict(X_wide=X_wide, X_tab=X_tab)`
	`167`	`+ preds = trainer_new.predict(X_wide=X_wide, X_tab=X_tab, batch_size=256)`