address review comments

Author: ndgrigorian

mkl_random/__init__.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without

mkl_random/interfaces/__init__.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without
@@ -25,3 +24,6 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 from . import numpy_random
+
+# submodules
+__all__ = ["numpy_random"]

mkl_random/interfaces/_numpy_random.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without
@@ -67,15 +66,39 @@ def seed(self, seed=None):
         For full documentation refer to `numpy.random.seed`.
 
         """
-        return super().seed(seed=seed, brng="MT19937")
+        return super().seed(seed=seed)
 
     def get_state(self, legacy=True):
         """
         get_state(legacy=True)
 
         Get the internal state of the generator.
 
-        For full documentation refer to `numpy.random.get_state`.
+        Parameters
+        ----------
+        legacy : bool, optional
+            Flag indicating to return a legacy tuple state.
+
+        Returns
+        -------
+        out : {tuple(str, bytes), dict}
+            The returned tuple has the following items:
+
+            1. a string specifying the basic psedo-random number generation
+               algorithm. It should always be `MT19937` for this class.
+            2. a bytes object holding content of Intel MKL's stream for the
+               generator.
+
+        If `legacy` is False, a dictionary containing the state information is
+        returned instead, with the following keys:
+            1. `bit_generator`: a string specifying the basic psedo-random
+                number generation algorithm. It should always be `MT19937` for
+                this class.
+            2. `state`: a dictionary guaranteed to contain the key
+               `mkl_stream`, whose value is a bytes object holding content of
+               Intel MKL's stream for the generator.
+
+        Compare with `numpy.random.get_state`.
 
         *Compatibility Notice*
         As this class uses MKL in the backend, the state format is NOT
@@ -615,6 +638,9 @@ def __NPRandomState_ctor():
 get_state = _rand.get_state
 set_state = _rand.set_state
 random_sample = _rand.random_sample
+ranf = _rand.random_sample
+random = _rand.random_sample
+sample = _rand.random_sample
 choice = _rand.choice
 randint = _rand.randint
 bytes = _rand.bytes

mkl_random/interfaces/numpy_random.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without

mkl_random/mklrand.pyx

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without

mkl_random/tests/__init__.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without

mkl_random/tests/test_random.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without

mkl_random/tests/test_regression.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 # Copyright (c) 2017, Intel Corporation
 #
 # Redistribution and use in source and binary forms, with or without