[BUG] missing parameter "optim_batch_size" in UMAP class constructor

[CJrZhang]

Describe the bug
"optim_batch_size" parameter is missing in UMAP class constructor. Which is mentioned in the doc

Steps/Code to reproduce bug
Init the UMAP class with an argument passing to the parameter 'optim_batch_size'

Expected behavior
TypeError: init() got an unexpected keyword argument 'optim_batch_size'

Additional context
Code Segement

cuml/python/cuml/manifold/umap.pyx

Lines 336 to 359 in 7e91505

	def __init__(self, *,
	n_neighbors=15,
	n_components=2,
	n_epochs=None,
	learning_rate=1.0,
	min_dist=0.1,
	spread=1.0,
	set_op_mix_ratio=1.0,
	local_connectivity=1.0,
	repulsion_strength=1.0,
	negative_sample_rate=5,
	transform_queue_size=4.0,
	init="spectral",
	verbose=False,
	a=None,
	b=None,
	target_n_neighbors=-1,
	target_weight=0.5,
	target_metric="categorical",
	handle=None,
	hash_input=False,
	random_state=None,
	callback=None,
	output_type=None):