File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change 93
93
sphinx_gallery_conf = {
94
94
"examples_dirs" : "../examples" ,
95
95
"gallery_dirs" : "tutorials/_rendered_examples/" ,
96
- "ignore_pattern" : "utils.py"
96
+ "ignore_pattern" : "utils.py" ,
97
97
}
98
98
99
99
# Setup the breathe extension
Original file line number Diff line number Diff line change @@ -347,7 +347,7 @@ def unwrap_tensor_shape(
347
347
if isinstance (tensor , int ):
348
348
tensor_shape .append (tensor )
349
349
elif isinstance (tensor , torch .SymInt ):
350
- min_max_opt : Dict [ str , int ] = extract_var_range_info (tensor )
350
+ min_max_opt = extract_var_range_info (tensor )
351
351
tensor_shape .append ((min_max_opt ["min" ], min_max_opt ["max" ]))
352
352
elif isinstance (tensor , (torch .Tensor , FakeTensor )):
353
353
for dimension in tensor .shape :
You can’t perform that action at this time.
0 commit comments