Skip to content

Commit dd45074

Browse files
Try skipping create accurate span bc too expensive
1 parent d76ef93 commit dd45074

File tree

1 file changed

+30
-3
lines changed

1 file changed

+30
-3
lines changed

python/tests/__snapshots__/test_array_api/TestLDA.test_trace.py

Lines changed: 30 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,38 @@
11
_NDArray_1 = NDArray.var("X")
22
assume_dtype(_NDArray_1, DType.float64)
3-
assume_shape(_NDArray_1, TupleInt.from_vec(Vec[Int](Int(150), Int(4))))
3+
_TupleInt_1 = TupleInt.from_vec(Vec[Int](Int(150), Int(4)))
4+
assume_shape(_NDArray_1, _TupleInt_1)
5+
assume_isfinite(_NDArray_1)
6+
assume_dtype(_NDArray_1, DType.float64)
7+
assume_shape(_NDArray_1, _TupleInt_1)
8+
assume_isfinite(_NDArray_1)
9+
assume_dtype(_NDArray_1, DType.float64)
10+
assume_shape(_NDArray_1, _TupleInt_1)
11+
assume_isfinite(_NDArray_1)
12+
assume_dtype(_NDArray_1, DType.float64)
13+
assume_shape(_NDArray_1, _TupleInt_1)
14+
assume_isfinite(_NDArray_1)
15+
assume_dtype(_NDArray_1, DType.float64)
16+
assume_shape(_NDArray_1, _TupleInt_1)
417
assume_isfinite(_NDArray_1)
518
_NDArray_2 = NDArray.var("y")
619
assume_dtype(_NDArray_2, DType.int64)
7-
assume_shape(_NDArray_2, TupleInt.from_vec(Vec[Int](Int(150))))
8-
assume_value_one_of(_NDArray_2, TupleValue.from_vec(Vec[Value](Value.int(Int(0)), Value.int(Int(1)), Value.int(Int(2)))))
20+
_TupleInt_2 = TupleInt.from_vec(Vec[Int](Int(150)))
21+
assume_shape(_NDArray_2, _TupleInt_2)
22+
_TupleValue_1 = TupleValue.from_vec(Vec[Value](Value.int(Int(0)), Value.int(Int(1)), Value.int(Int(2))))
23+
assume_value_one_of(_NDArray_2, _TupleValue_1)
24+
assume_dtype(_NDArray_2, DType.int64)
25+
assume_shape(_NDArray_2, _TupleInt_2)
26+
assume_value_one_of(_NDArray_2, _TupleValue_1)
27+
assume_dtype(_NDArray_2, DType.int64)
28+
assume_shape(_NDArray_2, _TupleInt_2)
29+
assume_value_one_of(_NDArray_2, _TupleValue_1)
30+
assume_dtype(_NDArray_2, DType.int64)
31+
assume_shape(_NDArray_2, _TupleInt_2)
32+
assume_value_one_of(_NDArray_2, _TupleValue_1)
33+
assume_dtype(_NDArray_2, DType.int64)
34+
assume_shape(_NDArray_2, _TupleInt_2)
35+
assume_value_one_of(_NDArray_2, _TupleValue_1)
936
_NDArray_3 = asarray(reshape(asarray(_NDArray_2), TupleInt.from_vec(Vec[Int](Int(-1)))))
1037
_NDArray_4 = astype(unique_counts(_NDArray_3)[Int(1)], asarray(_NDArray_1).dtype) / NDArray.scalar(Value.float(Float(150.0)))
1138
_NDArray_5 = zeros(

0 commit comments

Comments
 (0)