version | 0.0.21

Yura52 · Oct 10, 2023 · d445999 · d445999
1 parent ba88e5c
commit d445999
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/delu/__init__.py b/delu/__init__.py
@@ -1,5 +1,5 @@
 """Deep Learning Utilities for PyTorch users."""
-__version__ = '0.0.21.dev0'
+__version__ = '0.0.21'
 
 from . import cuda, data, hardware, nn, random, tools, utils
 from ._stream import Stream

diff --git a/delu/random.py b/delu/random.py
@@ -87,15 +87,15 @@ def generate_state(*args, **kwargs) -> np.ndarray:
 
     np.random.seed(generate_state(4))
 
-    torch.manual_seed(generate_state(1, dtype=np.uint64)[0])
+    torch.manual_seed(int(generate_state(1, dtype=np.uint64)[0]))
 
     if one_cuda_seed:
         if not torch.cuda._is_in_bad_fork():
-            torch.cuda.manual_seed_all(generate_state(1, dtype=np.uint64)[0])
+            torch.cuda.manual_seed_all(int(generate_state(1, dtype=np.uint64)[0]))
     else:
         for i in range(torch.cuda.device_count()):
             torch.cuda.default_generators[i].manual_seed(
-                generate_state(1, dtype=np.uint64)[0]
+                int(generate_state(1, dtype=np.uint64)[0])
             )
 
     return base_seed

diff --git a/docs/version.rst b/docs/version.rst
@@ -1,6 +1,6 @@
 .. this page is needed for themes without the version dropdown
 
-v0.0.20 ▼
+v0.0.21 ▼
 ============
 
 - `development <https://yura52.github.io/delu/dev>`_