How to use the pygw.base_models.DataStore function in pygw

"""
        Create an HBase data store

        zookeeperh (string): zoopkeeper for hbase
        hbase_namespace (str): namespace of hbase
        """

        if hbase_namespace:
            j_hbase_opts = config.MODULE__hbase_config.HBaseOptions(hbase_namespace)
        else:
            j_hbase_opts = config.MODULE__hbase_config.HBaseOptions()
        j_hbase_opts.setZookeeper(zookeeper)
        j_ds = config.MODULE__core_store.DataStoreFactory.createDataStore(j_hbase_opts)
        super().__init__(config.GATEWAY, j_ds)

class RedisDs(DataStore):
    """
    GeoWave RedisDB datastore.
    """

    __compression_opts = ["snappy", "lz4", "none"]

    __compression_to_j_enum = {
        # Key: [compression_type : string] => Value: [function : () -> Java Ref to RedisOptions.Compression Enum]
        "snappy": lambda : config.MODULE__redis_config.RedisOptions.Compression.SNAPPY,
        "lz4": lambda : config.MODULE__redis_config.RedisOptions.Compression.LZ4,
        "none": lambda : config.MODULE__redis_config.RedisOptions.Compression.NONE,
    }

    def  __init__(self, address, gw_namespace=None, compression="snappy"):
        """
        Create a Redis Datastore. 

batch_write_size (int)
        """
        if gw_namespace:
            j_rock_opts = config.MODULE__rocksdb_config.RocksDBOptions(gw_namespace)
        else:
            j_rock_opts = config.MODULE__rocksdb_config.RocksDBOptions()
        if dir != "rocksdb":
            j_rock_opts.setDirectory(dir)
        if not compact_on_write:
            j_rock_opts.setCompactOnWrite(compact_on_write)
        if batch_write_size != 1000:
            j_rock_opts.setBatchWriteSize(batch_write_size)
        j_ds = config.MODULE__core_store.DataStoreFactory.createDataStore(j_rock_opts)
        super().__init__(config.GATEWAY, j_ds)

class HBaseDs(DataStore):
    """
    GeoWave HBaseDs datastore.
    """
    def __init__(self, zookeeper="example", hbase_namespace=None):
        """
        Create an HBase data store

        zookeeperh (string): zoopkeeper for hbase
        hbase_namespace (str): namespace of hbase
        """

        if hbase_namespace:
            j_hbase_opts = config.MODULE__hbase_config.HBaseOptions(hbase_namespace)
        else:
            j_hbase_opts = config.MODULE__hbase_config.HBaseOptions()
        j_hbase_opts.setZookeeper(zookeeper)

from pygw.config import config
from pygw.base_models import DataStore


"""
stores.py
====================================
Data stores
"""

class RocksDbDs(DataStore):
    """
    GeoWave RocksDB datastore.
    """

    def __init__(self, gw_namespace=None, dir="rocksdb", compact_on_write=True, batch_write_size=1000):
        """
        Create a RocksDB datastore.

        Args:
            gw_namespace (string) : namespace 
            dir (str) : directory of datastore
            compact_on_write (bool)
            batch_write_size (int)
        """
        if gw_namespace:
            j_rock_opts = config.MODULE__rocksdb_config.RocksDBOptions(gw_namespace)