TypeError: ImageSegmenterOptions.__init__() got an unexpected keyword argument 'output_category_mask'

[saboorniazi]

import mediapipe as mp
from mediapipe.tasks import python
from mediapipe.tasks.python import vision
import os
model_path = 'data_gen/utils/mp_feature_extractors/selfie_multiclass_256x256.tflite'
if not os.path.exists(model_path):
os.makedirs(os.path.dirname(model_path), exist_ok=True)
print("downloading segmenter model from mediapipe...")
os.system(f"wget https://storage.googleapis.com/mediapipe-models/image_segmenter/selfie_multiclass_256x256/float32/latest/selfie_multiclass_256x256.tflite")
os.system(f"mv selfie_multiclass_256x256.tflite {model_path}")
print("download success")
base_options = python.BaseOptions(model_asset_path=model_path)

options = vision.ImageSegmenterOptions(base_options=base_options,running_mode=vision.RunningMode.IMAGE,output_category_mask=True)