add a function to set random seed (#17)

* add a function to set random seed

* changelog

CHANGELOG.md

@@ -2,3 +2,4 @@
 
 ### Features
 - Added exporting the graph to `scipy.csr_matrix` ([#16](https://github.com/AleksanderWWW/abcd-graph/pull/16))
+- Added a function to set the random seed ([#17](https://github.com/AleksanderWWW/abcd-graph/pull/17))

src/abcd_graph/utils.py

@@ -19,12 +19,14 @@
 # SOFTWARE.
 
 
+import random
 from functools import wraps
 from typing import (
     TYPE_CHECKING,
     Callable,
 )
 
+import numpy
 from typing_extensions import (
     ParamSpec,
     TypeVar,
@@ -37,6 +39,11 @@
 R = TypeVar("R")
 
 
+def seed(num: int) -> None:
+    random.seed(num)
+    numpy.random.seed(num)
+
+
 def require(package_name: str) -> Callable[[Callable[P, R]], Callable[P, R]]:
     def deco(func: Callable[P, R]) -> Callable[P, R]:
         @wraps(func)