Skip to content

Commit ec22aa8

Browse files
committed
Breakup string
1 parent f4d5f44 commit ec22aa8

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

test/utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,10 @@ def get_frame_data_by_index(
392392
if stream_index is None:
393393
stream_index = self.default_stream_index
394394

395-
tensor_file_path = f"{self.get_base_path_by_index(idx, stream_index=stream_index, filters=filters)}.pt"
395+
base_path = self.get_base_path_by_index(
396+
idx, stream_index=stream_index, filters=filters
397+
)
398+
tensor_file_path = f"{base_path}.pt"
396399
return torch.load(tensor_file_path, weights_only=True).permute(2, 0, 1)
397400

398401
def get_frame_data_by_range(

0 commit comments

Comments
 (0)