jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed type alias

Browse Source
This commit is contained in:
jens
2026-06-03 22:13:26 +02:00
parent c62605de51
commit 27460c503d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/matrix.py
+2 -2
View File
@@ -3,12 +3,12 @@ from typing import TypeAlias
USE_CUDA = 1 USE_CUDA = 1
if USE_CUDA: if USE_CUDA:
import cupy as np import cupy as np
Mat: TypeAlias = np.array Mat: TypeAlias = np.ndarray
def convert(src: Mat): def convert(src: Mat):
return np.asnumpy(src) return np.asnumpy(src)
else: else:
import numpy as np import numpy as np
Mat: TypeAlias = np.array Mat: TypeAlias = np.ndarray
def convert(src: Mat): def convert(src: Mat):
return src return src