Viewing file: _pickle.py (2.26 KB) -rw-r--r-- Select action/file-type: (+) | (+) | (+) | Code (+) | Session (+) | (+) | SDB (+) | (+) | (+) | (+) | (+) | (+) |
from .mtrand import RandomState from ._philox import Philox from ._pcg64 import PCG64, PCG64DXSM from ._sfc64 import SFC64
from ._generator import Generator from ._mt19937 import MT19937
BitGenerators = {'MT19937': MT19937, 'PCG64': PCG64, 'PCG64DXSM': PCG64DXSM, 'Philox': Philox, 'SFC64': SFC64, }
def __bit_generator_ctor(bit_generator_name='MT19937'): """ Pickling helper function that returns a bit generator object
Parameters ---------- bit_generator_name : str String containing the name of the BitGenerator
Returns ------- bit_generator : BitGenerator BitGenerator instance """ if bit_generator_name in BitGenerators: bit_generator = BitGenerators[bit_generator_name] else: raise ValueError(str(bit_generator_name) + ' is not a known ' 'BitGenerator module.')
return bit_generator()
def __generator_ctor(bit_generator_name="MT19937", bit_generator_ctor=__bit_generator_ctor): """ Pickling helper function that returns a Generator object
Parameters ---------- bit_generator_name : str String containing the core BitGenerator's name bit_generator_ctor : callable, optional Callable function that takes bit_generator_name as its only argument and returns an instantized bit generator.
Returns ------- rg : Generator Generator using the named core BitGenerator """ return Generator(bit_generator_ctor(bit_generator_name))
def __randomstate_ctor(bit_generator_name="MT19937", bit_generator_ctor=__bit_generator_ctor): """ Pickling helper function that returns a legacy RandomState-like object
Parameters ---------- bit_generator_name : str String containing the core BitGenerator's name bit_generator_ctor : callable, optional Callable function that takes bit_generator_name as its only argument and returns an instantized bit generator.
Returns ------- rs : RandomState Legacy RandomState using the named core BitGenerator """
return RandomState(bit_generator_ctor(bit_generator_name))
|