@@ -299,11 +299,15 @@ def test_value(self):
299299 generator .manual_seed (2147483644 )
300300
301301 x = functional .random_hv (4 , 1000 , generator = generator )
302- res = functional .multirandsel (x , p = torch .tensor ([0.0 ,0.0 ,1.0 ,0.0 ]), generator = generator )
302+ res = functional .multirandsel (
303+ x , p = torch .tensor ([0.0 , 0.0 , 1.0 , 0.0 ]), generator = generator
304+ )
303305 assert torch .all (x [2 ] == res )
304306
305307 x = functional .random_hv (4 , 1000 , generator = generator )
306- res = functional .multirandsel (x , p = torch .tensor ([0.5 ,0.0 ,0.5 ,0.0 ]), generator = generator )
308+ res = functional .multirandsel (
309+ x , p = torch .tensor ([0.5 , 0.0 , 0.5 , 0.0 ]), generator = generator
310+ )
307311 assert torch .all ((x [0 ] == res ) | (x [2 ] == res ))
308312
309313 x = functional .random_hv (4 , 1000 , generator = generator )
@@ -325,4 +329,4 @@ def test_device(self):
325329 assert res .dtype == x .dtype
326330 assert res .dim () == 1
327331 assert res .size (0 ) == 100
328- assert res .device == device
332+ assert res .device == device
0 commit comments