from typing import Union
from deepchem_server.core.datastore import DiskDataStore
_DATASTORE = None
[docs]
def set_datastore(datastore: Union[DiskDataStore, None]) -> None:
"""Set the global datastore instance.
Parameters
----------
datastore : DiskDataStore or None
The datastore instance to set as the global datastore, or None to reset.
Returns
-------
None
"""
global _DATASTORE
_DATASTORE = datastore
[docs]
def get_datastore() -> Union[DiskDataStore, None]:
"""Get the current global datastore instance.
Returns
-------
DiskDataStore or None
The current datastore instance, or None if no datastore has been set.
"""
return _DATASTORE
[docs]
def refresh() -> None:
"""Reset the global datastore to None.
Returns
-------
None
"""
set_datastore(None)