From b72dcbf103781f1ee1e0ebf53220641f31ff72c9 Mon Sep 17 00:00:00 2001 From: gamzeisl Date: Tue, 24 Sep 2024 18:35:24 +0200 Subject: [PATCH] Fix python format --- PyITA/util.py | 1 + 1 file changed, 1 insertion(+) diff --git a/PyITA/util.py b/PyITA/util.py index 14397c2..958b99d 100644 --- a/PyITA/util.py +++ b/PyITA/util.py @@ -23,6 +23,7 @@ from numpy import int8 as i8, int16 as i16, int32 as i32, float32 as f32, uint8 as u8, uint16 as u16 + def random_shuffled_tensor(shape, bitwidth: int, type: DTypeLike = np.int8, scaling = 1 / 4) -> np.ndarray: """ Generates a random shuffled tensor with a specified shape, bit width, and type.